/*
* SPDX-FileCopyrightText: Copyright (c) 1993-2026 NVIDIA CORPORATION &
* AFFILIATES. All rights reserved. SPDX-License-Identifier: Apache-2.0
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/


#pragma once

#include <flashinfer/trtllm/batched_gemm/trtllmGen_bmm_export/BatchedGemmOptions.h>


namespace batchedGemm { 


namespace tensorrt_llm
{

namespace kernels
{
// clang-format off

#define TLLM_GEN_EXPORT_VERSION "7.0.4.0.4.0"

static constexpr size_t tllmGenBatchedGemmListLen = 6966;


static const batchedGemm::BatchedGemmConfig tllmGenBatchedGemmList[] = {
#ifndef EXCLUDE_SM_100
{nullptr, 0, 215312, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm100f", 256, "10629e37fbdded20a51259eb5824a7ae69db984feaf8ab7de02a3e9b737022b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm100f", 256, "11148feefa23184ab6d3a444e12c8e1514f9f8082a266f9e19401051a37ee241", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm100f", 256, "8f0f06b9ad9b9d17b341306e20a0577cc41137fe00e1fd2994d152a973984bf0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm100f", 256, "69318df3abfa19457d9e5d2bec2ce112822ce031e71a00c7f709af5d5cbfdbbe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm100f", 256, "491f41de4001a31c75f15c7d15bd547054fde306fecce1ce5c83169c222d0a8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "60ed7ed4f8849a2b02edc391f59049d62bf886767d98c6434194a98ab45fdaf4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163344, "bmm_Fp16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm100f", 512, "b58f2a8cad736fc8be8773ec4023eb70811a6446a0fcb8377c1f952ecec6de42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163344, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm100f", 512, "c168aebaad4082dd95dc4665ee9852a21ff00f351dc73934e65760d294c437ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm100f", 256, "d05af7f6868f276360b383a24b1dddee368d5c2eccd8fddbdcee60d82985f867", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm100f", 256, "c1c2990a8f561783034469a0f6cb88f7861f046a85dab2c1734969bb3488541d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163344, "bmm_Fp16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512u2_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm100f", 512, "86a0925dadb5df51c933603abefa6ef816e6df943f90aabbbf26170f2b8371ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm100f", 256, "53b12f75503d9fa9581d0d7cf2169aee65ab812e8f08992d117c203d3d41f3a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm100f", 256, "838f1bd130d4b5a3f9f2300c6b25f44cde63c7ce367d5f74293d0c9038aa3b63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm100f", 256, "c7584a0aa112d61988b0e3e3701dd38a8fcfe2da2e1d4f661bb264f2c4ebe2cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm100f", 256, "425b232cd6f4da8793560373a79116c46f0faa84cc37fa3726ee4c36db41bab0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm100f", 256, "57d3d1a5ee052385f5ba0877da34627b004163f67250aa14fa496b377f71e271", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "35931bbc3b752d2eeafd4d3366f82ea9dd94b07535b85447d1553cc4ba9d4a3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "68b900da5fb8901a8168d8f3f8b32267ca0922f2c98e1e122864e95336ca81e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "3b6ce942e0157ba93bd18b7a0bb30b91fa4de0749240d7faccaf0066401db962", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "e5844e2896b913b281451352f2627648d7a5a8c6cf59c9c6067dfc427f337448", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 81696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "d9ecb266a3219221756d13c908b2d1d69ef7a98aefc3dfafc8ab011c3b280ba8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "94e76d51d36de331b438bedbb6ab0ed5663cd2a82adf4ddc7d503dc8e5a53a56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "6f3ac7b32fc18ba4f5d6ef1a190039fe1fd591574bc0ba7cc92ba4ae375954d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "342cef4106e168a9c96064e7f94cb2a1b9ae63d2209da768e25b98f9c3d8a1ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "c0945ec39a4833b214b3644c67a959fa998782a3b8efffad742ee1b1a69fce9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 81696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "d938f1cb0a4f8308978d38df7f470026406444721f1dcc15f58922cd4bbf757b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 81696, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "ddb4754828e6e43af8d89cdaadb62f77033cfb7dfafdfbb2d6e03419a691bf10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162320, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm100f", 512, "1f41e0bb03792f66a5b1641ecfd7f8cd6e1059e86c5abe43cd6c53ec8afc0bea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 81696, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "0a15a45091eadc7d831cb5461855d64293fbb36e6819c530b69405c08d187187", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162320, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm100f", 512, "94a198fcde9293a1619eca9040bc223b2ff1c570bd85b395d5607dbefb030faa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 81696, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "09797403879a5eda259c86b753d540d71e9dbd8ab703aae9dc966927e0a1eb8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163344, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512u2_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm100f", 512, "0ca82d9f09b42a76e2c9ad9ab5ddcb01aded5786db02c251df8a135bb882485f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "ba7345fe7c3c4c5122c29fbe7adcf885b13027376293c71b208f08b024d6ca44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 81696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "c2028d02e44e9f84b400d8001173d2e58d4372f02aa5910cc34f2ca5fa8b28d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "85d1cb3347ef353d6ce436ab7a0dc619dceb0c7f601b63be167c00b73f3a7a9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "8b39a4380fd318388d1c1f4bfa9a7e6563f6328483b5b2e9c9b189531b2c631e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "5b711e265dc42a2349e16713f48e1a2d32d5c716c3485c94657b0f8b7c9d0f47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 81696, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "379725211acede40f7304499ce930fe346df1df6fda431ef9a34f22cf32b655b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 81696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "f9a1dacbd3683cb71f9a4ad658123f18d7c5c634d1fffba8ea29629050b7b8fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "c0458caede47ca11d999747fe010712e51308d7367d4909376157bb41ba1c6e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "cbca0f08cfe4eed28ddb3487281699a8e247113c506272acbb3f58b49324c428", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "3dd0b7779ef5fee9dd38129e24c86699ffcdb45eb23e4965fe05b2b0d8f3d5da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "d0575ac341fd7931a78358fbea8ffae4feaa3d8ae23fa4d6c8dab664345d0b11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "5ba4b6736b6ecc6f9ecf6d5999d281b5ce36faa73ebf89760be6d5426d826cbf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "6bfa0489d9325351e114917f9127573cc2767e9cce8ced37b40925af4fe4d7ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "12209c1699cc5969c786707c857f3326517f815623cb0ec0dda81ef31865ec91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 81696, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "ca061df94010c1cf90d16a65d26e818f2601b6714c7a231224c3835d267827d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 81696, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "dc72b9f2f83a3135a1e76495845fe12ea6a922f3c31e22f9096d5c0bc9dbf5a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "12da2a9d2f9b0691d3fb70d3199eb1f6fd668cf3c8d3f3fa1cd7faaa20a7be98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "c1a2673172a13eb10141659eaa860e9d502dbb69ed711608e5a5bf2ec156ec41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "b1897c6fe79a940c87492c77dc47c87a9096173a67ceae3e1465c4551bff552e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 81696, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "9df4ad525c6a6e7e94276bf2df520069581e806a845450e72ce2da29ebf4e074", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm100f", 384, "38ba1bbf4deaa4577122cc05b3c1793f07db401718af3cf71078161e61617a9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 81696, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm100f", 384, "89aee49f4764c179e6b2e65a58652536227ce86cd09300d64bfde5f7a8828a2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0b5c56b36073e596d276e51518e40aa1b935c67334275d49506fa5ed746a86c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d83e5191dd42a169480f4835cbf6eb671f4b4a0a4114a7bc6a1e5a7e42c1595c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "92a924b11e3a5d2ddfa65a397346ddef4494d73df0a0dad87ee29e1bbee718b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "2de0bcb1ab87ef4758fb648187fc626c101635b1a053cab96eaa7f9387cb9876", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "44d0dbcb0b950a8b61560e367b49e201cf17fb76c00e9887013d98fe48aa35e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0597ee9704fb5092eac5d10deaa56510b1a6e54245758341cae01ecdd59b4a87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "24a5bad3c06a6764ac0a21de5328704a6a15ee86195901f1e30fff4fbde33c7b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "7f3e27e748b9c3298f9981ed62db7ae4e030587f4f196d1746349a5469295ffa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "82029a444ba3f15cd84bb760565554720ad92989f81c6626f50b839058263474", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ced3268366aeff2e49891e4416e4897a7c768c5e2f7f6544031b61c96125ffac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "769002c5fc5fc41c747d2893c4c6d13906917338790fd8b42c56dcba30e72b42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "cf70149bdfe3cf2d9ed21a1760f0634a23bb62b85adb27a36740b54d4d391d4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "d952cfa216875223a52c25ec47c3b697419ebd56d03b41b7a63a7b873a9de565", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5174b853413d07a80eda3583a5bbfa9eaad35443e2492dc46116bc9b754a9c44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "39c9f12332fbf7f7ce57b0da92df20a7fea3cbb9be6e884bd16fd27e9f851f21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "44b712ae303eb47f3e68c9335dec895d643b1b54a5ff94e0766b89b26eeb5f0e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "263530740cf78224ca24c729b32380e51cf8d740ca75201aa7bc006179456c50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "fe393541ecb1820629d05bf73eb983f91c6f9bfdc7a8154cb977ae416a4806f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "eaaedd7d20dccbcef6d163ab3e393e9d7ee76a86842800b9c09e7cc2f9f6aff5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "479c9ef6f6596cd45b3034df2314453d5c34f59d2d410a55fbe9d41ccd16e718", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "bc7eed8a8e2a5a9877b1d5c9bb3e3a564bc4a086d8b1db42aac29b735d0c830a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "632b07ecad5a9c54cd3d9cc8fa71a08e3c10b727d11c9ffd43d5c264a72c327f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "07d9b76e4ca8ca33a54f10aaa0737ba62e07f6106bc6a81d8205cc967eabfcc5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "414592db3c090b7ef56a3e0461af9d710dfc9dda526af314d2bc727cb32671a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "90544d63d57c1bef3454fe46ad2956535ca4ce81f682511ba546c93f608bff64", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "6b1e8954afef12866da3733a5bf6e553b53f9c17f9ee5ad0ef01033a70997394", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "b7bc5ec192c4c1167ba3f393e2414d1776157d0a565f1cde7ad1c054c0ab4e77", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "1896c741761e5cc88f4fc12cc6c786ff8f470f3223d9d53b58ed417b9a68d6bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "3f6e0398ba57c52ad4a3138d831018647de2009f8751bd874ca7f0a8261c4bc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "95987408e9416ba93042528a2760f8880e7b1be024787abea1dc828746e06f90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "27fc6516990dbc89983ada273a829f8056258b1b33845de51beacfb238603267", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "9894e5e3cadb3a8eb76ee9056094e255b38fe489be2d1dd496503c08703ce82a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "7bfceead12d68523bc576a35e80a1ff02a0faa91be6713499c6833a35acee7aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "79a70feed0317f6d789d06fae5a6a39ed55353a38190083d4e19fae05aa61648", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "bafaf1475fd7c375152ae3ec897f6a5af2d445b7288fcab409889c600a116fb0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "3db98f7d1aa56477ed3b76758f3c6aa61eddda9f2b493234097c79a11a02432e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "8efbfc3bdb28d490efa7d391440b50f58e20637f2d6ed2db61eaa59624bbac24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "9c37fee436728fb9f94bde3f9d841c131252a28db452f97b12b351f7ea0c1814", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "5446a7f9cadae49b4eb8e70967742b7f66ac0c7221834ff3cf19884956fff919", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "58555c050da1c80ba447ffb1faa9df8fce394b9a0cfbcc06ec4cfd28880b2adc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "ff6051b63d9586e1539fa3db0f2bb5ebaba562404aec91477ef01ec58bdd1193", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "00ce6561d9643100442608897dab12494b0a142acc598679c3de63680ffe6fac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "1e906881f82a5cc15ede77d70746f75ebd385964f426c556d3cafd342c6d1389", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "9301916e271ebeac7313dd99b195457ed02a9d9874121e600e7c7afafa562064", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "79c7fa501b43bc1edb35b4f603a3ef8dcd569822e39aa439dc7028d3d1a5eb22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f7bf843f77e8ebfb99911152435da749ce30ec0007413bf9da6f81906ce4d9ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "aeeb65307dfda848711bff47d76fe5f91968896a2df93689f5f25b015d53e909", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c5cd98f8284c7246ae4d2cbb215bb26a3e7162e42055bf9ecd38f1e0947432b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "3f834f05770bcd2393fb76bc3063e7dbc9a02250ce587c142e6517cefd48d219", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a402b5e27adb423a46caf55b77c722287e7641935c987bb82e2138ca96d281ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "c6e06db76853bcef67120a83ee5e8e6128c43c276c51e611f7c80d099f2950cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "033d7091327066559f55253c05645b3b2176b256899f21ea0bc4a86c24b4d257", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "a9fb371718069e33606ad292c7a46dbd7d1060c9f562836ca7b895afa5f2dba5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 198160, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "451ffb915276cc3da9e9054deb68b9cbaa2d4be12034dd309f8e7ae87f299886", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "838c5af9e36a857bb212d254b6cc84654e8e8da308b5e37d25a49d64b67ce0e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 198160, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "b004be9d40eb69f16787ef0d1c152e70a2b190f40da66b6362252d78266b3ad2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "deeb4bebe7c7bdba47c6bf061f2ae4c3517b5156c87d32c545231ad886540005", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "d687484e5ff4161cb94b53bea3a4a1f30ae2538c4bedd0c222ed15d692c17899", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "bf57fdd3a6afaf6675a5efc992a87f1eae97a6565eb5be78e676d0424ab960f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "684f7ddb0e49ff1e69ed140806c790771e56d9e2def675672ea6a4af3e55efa7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "85ae6d6cb93f6e7254e2a66b452b225cadc2777c4ffd0fe386270070eec59b5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "77787e575719019fbc94054efb94b2e2d0b02dbe5fcfff4474d3bd22588b5bdc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 198288, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "7d65c30d17f09d1a3d2ff96fe5bcea8ba0abfd4fadb844569ea736036e663bae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "7c404c1eba41bafa4bcb5e5deec8d1accf25bbc3322d7c89b68d8d12e5e76374", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 198288, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "b75af4adfa891618abd2ce70d1c34c09ef666f04a3c2b887972ba721102f03d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "cd4ffeba85c21a0cae356bae04279bfb901c308d893f6e4738ca690d944a2eb5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 198160, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "8b25b43a4ecc74c4058afa5c3c7c0936f95407dc4cddfec98d8abe24cf52b1d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "6a690a91385faa664282dd811c9ff8cac933e4e806d3eb59deb9bcd21ae8809f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "81da5a7ebe9afc55351eb553a9a85cde4b9982ff90c9d1fab6ed7f4eaeafe8d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 198160, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "81c10e9287690cc8ca6ce9640fde3bdff47c58d67f53adb38e23c3ff198289ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "486647ca8d87f77d607defacb16dc79b3d31466ac7f3aafe9ed400f94137311e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 198288, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "e0686c829c5f9f5372ab23d2121781d5b5894b355f9dcf908439c4392bb66e59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 198288, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "302e148ffb09e8bd1c9f4a0bdcd8faa831bf202b03c04f14422e8f9c012d2d53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "00fdab1c7627525f0f82b0614d316037af656acf791e36937f915523ec4939a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213520, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "558537df83e9a1c0f352461bb5bc94a26eb6f785ca8d570e3c40871484ede471", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "154b9ab80ec91cd71aadfb733c6a0ddd5d3bd8c36701e116ffa9e52c18b064b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213520, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "9dccf957495d645ccaba4d0fe6a3d14c93f1b39c21e42e1fd0070b3d16c5a0ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213904, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "42f50b38415100ba407285f627a9c2f356e9590321efe2905752d78bcc9922f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "188fc55fff3b08b534783d379a309edc272859a75db2fafd11d9ca83e839dfb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192016, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "cf566876e14c86b03908563b94b33cbb2ae24c94ef4bd7bf5837bcef92d9e5be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a4f8464211c9fafbac30d5bddbe810d98c3a7cce82269550c586c6c71d9254b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5edd62a3f358f3f3e22f1a6c16b3aa92a178ffe5cab604d64f7f697dafa4c291", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213904, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "35c64d4d7d08e5436e15b0a6a5995679ed8751d2753c82f65ed93757c3413aae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192016, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "199d6f16b8946e428bf45847c946905c92429a4f7d683be3934fca60dfcf7f4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e1d60b35f4a82c7791c2453fe499a894f94e39204b0fcfdb705fe7ed90b9e7f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192144, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c89c3227ac543981adb4058eb12e0a40d28a229e49486d3753336da94b76eb0f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "e3ad5b38d75a284e7dee1f5c7454995cfa907bcffa8c6bd32a4af1783a42b068", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "57a5d2b2181269c37e1df7c0fa27e5f4e721523876a08ca2f4c97c0487615169", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "094167743ec7c87b344a604daf37944dcab7c3569d112f9e8abdc4961d048a48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "4986406eace771178a9add09fd74c7990a2cc339178ef0d70b471abc15e51751", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192144, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c9faf95d1781d87e67794e3eef2e04ea976bf274dd3db19aab8a44b804f1844c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "1a7d0f733b69efd5af27660f49653f409a348b1b36f1fe3858272503303ed239", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "09dfedcecedf9a0b20f9e8b0c4bd015297df9330ff0ea2ba2f9103e6631574a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e6cc758c88b4f14f9901e6ed5821dda01ff6d7bbb4d26072c75917b15bc13f9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "fabe63ff380b2ae474c96d4be5fb18fbbd78d12d0719f08596495596d7c87357", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "fbaab6229e8f93693a7d46b61afca8d7f7f8228d8c2c3e6ffec22e6dc081f067", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207376, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "80abeeeac5c36c2c90322e260e72c9278ad69809a79fab6fa37d0fa822ab6116", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207376, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "fcce619c42a1673022b578ffb16768886120b50edc28a8db4382018665fd3633", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "2466997ed15ca7ea800ebb0524ad23b7f15094d0ec4272e788a813f5dac3a320", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "4a8805c72049a58938784959e29ece48de38fcd25bb049987e66fabd2dbd9399", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "3ad4587ec10e4b2e13b6adc98f178ea24934bbb3aa012494369041cb9399bca1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207376, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "55404ad6ed633e19f919d2b76e6b5a2dc1db8c743754ae5cdeb5d5bbc1f58869", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207504, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d7075be8ada378d7752313e9b2b481e497eeb5a9076d17f619635db06902f12c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207504, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "4f3f949332f03d6b608f2d8f65dd851b45d642d46cb523e1c8400a32033289ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207504, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "4f9c4dbd41afee91f47c52db00b1a61557a5013cd04e0f5e60eb1a3b0c12a153", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "a01c789ef0383e5507bfd9dbd059bdef74fca8281ed2c7a55304927cd0b0d981", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207376, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "749a5ba2a547ba2e737a6848ead17b373cfa3bb11229a44627f730df5a35426d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207376, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "4c63bc19450ae4d90aec3fe7312ecd859b32c1ac9876b01e8a65c2b74fb06aba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207376, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "13876030effd438aee6a92f321e9966997f7f5024e469f1b8a606ac3af4c1c8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 224784, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5d0305037c376394fb98f80ee14182e32d5eff2748129c3623a5f862c4da3a47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a8f00d52a74c32b613971cd1180fb696fb1447b98a9b085f0d7c91e64847d179", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5b42ac2cab5e660145f0594db021b3dbe8b04f6f12c77bba8eaef2dd85ab160b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5785ae4dbdf0e80844b5efe18eec10b190a1c01407e148a023759f3150c59a96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0de81f15f504799fa628a5451d6c403271ed19e8211fce8afdfc1f4e88147a65", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 224784, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "da7d0aa90ac6741bf475d219610762d7955e4b5e2693fc11d620aff3adf06513", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207504, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "5aa2b8e9f40511a09b53bd0faa7b8dfbd276d2c97794b07096c5674cda5212d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225168, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d1788b4dea6ed71ca35c2b7c2f4173b58fa5601dba59c761af87eda7624d59f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213520, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0b7be91ee2fe0f2f55f8169a31d45e3a323e2cc6634485210888922dd0aef7b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "0954493661602d45a0523d5e15787462fba7406d04f4e2046bc38e846afd086f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "95a1113daa5e979c4647806acf993be259a92cc134cdd09a95d393bf55248c8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213520, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e48a321413c6f012bb6dcd92ddd8790b1556a8a9b07b4df2cdc3bc75413a8a1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "6819de3b99d774764750a8afe522e3724fb4b188f8186f2acfe523d094637c6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225168, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "92a76309c369a3e52c6b007db4217b2c7c62d1953a6faef379a3744e68dd8e10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5fbd921afefd23be8480614831a640929c12541dd43fdbf9ae9ed8604be7dc69", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "450b0a01ffc3b8f5a750831adfcd27ee010f7cee8e8e931680edd4f6616705f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213648, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "80a5bbf167758b0590a77ce29bc3bd1c2b93e9db2c1daaea7193d7359fae7aff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "babc0323255f0dcd96f8825819dc6b08a53c369d02b036a0cb395545b9f1c525", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "d41e119b4040a5c4171fcfdac668bed6764f0477b00f90ec692b4d09b69a96d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "770de4cfef1b5b1240d358b43899311d64943397fc02a66616fe351704158f0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "5e936f4d2cb01f37b87367c433b61f4da82e2e5d9806fdf248e5569369f9cce8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "fad36dbc363b0faf86fe6dc3a795ac7f12e9a4aeb8d9dab48ec318f07880f795", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213648, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "de63e072a918275e69197bffcfccf74f7159ab2fccb72b51f5885a3b6c6b5084", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "aa484af9b628a485d4a40ef180e4b09ba278eff7d12e0b6e64ea65376fb1afa4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "3ac06486346d890760b405bb29fbfce8a33a7d7c8e0edfeea053ec658de5e490", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "9661a9f620d5d07400feb25028d8f0b7a0652ca6ab0bcdc864e2112ee11f103c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "993f19dd1983b18e3ca4a4675b179ce4885f8ac5667bb60fb677fde7edcdc2e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "fb5f8e8a05f49a9ae24456856ecf36324c5104f2e9ff213988386871af1ab92d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "0d54024d706d8058e9efb1baf55b211f9af83f708eae074c3c6e429c50458513", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "1de3a6c018f6d6e05caba30915a3590985a83b8f6e8d6c87248cd6af3e764ec9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "13a0e648e27450ee0a885d4e0519d8ad91be0121b6d00c8ace957f2537a11bc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "7d07ac865e4f1458dedc0fb5d3ba9d8be4eec7a9fa8ed70410161d08a7907677", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "8873a5af408a5c979306ff81f84d430c7ff65987a26c49b677ec33de8f8d7d30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "c707a6812a827c3b12ef8e1a2063e3023e8efe43386c42f4c3cebf68fa181c8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "9fe9f7e1e8f4a5a64ea141004bc0d437a99c76f00662af42c394c65ab81e5c44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "a5afb94957e0d1feb963273c9350f5bf3cb928347e3f376b23906206d654117d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "ae950b5b675e1b376d02b3a0af333698b4bac56de6d3aa3ec582f183df0be73c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "0ff223923b896680b5b25eff4923fc51dd1ef775fe9cecbed2dd8852f128657e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "f9ce91b7fed74802d547ad72b5c667bebb1d4881827d4c7fb237286494255645", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "3a068ad4a238000b01c99d58f6b23da9440bcb8c367331f6231e9dd6d149efc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "41f0961e5650c15b6d1edcdd2023a8cd7bf0d35feabecab850fdd273faae6a15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "fc6824faa0260b4d0e70ae0f9395530d93b50d208744bf9db59b86987d889ef8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "6c587abbe5dc814421bbc6e4143c8ad495ef34bc523ff5ac46b49de50d5cbe74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "bbaaefa7435789bdd8e23a702727caa0b2e07e024178e6fab50693d8ce6e5991", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "eb672672c036798c973ad01317d1cfee1ec4096ceeb3d6964046a5b6937718ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "57bf1f275ea6d28cd1d3700c06596268ddf1e4e2d3a9b942a565dd39a2099c96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "422828408f10ee09a6004a1c26fdbd289d8174850997c495105c0db256423cb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "f44512d08b9216fee8b61984d5210be461cd82c5f576d935e35b6a8ea78e8d9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "5e828fedf367c10398c12c20172267bf36ced8d93e6d9197d6301a1b61fa96a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 768, "9e77375e31dfe9b0146b4a93bf59c0687aa19309e703286efbe93a623aa3a66c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "8c6439198fb71c80f8b7e4eda13a2b66cd6dabb665b6a5f407860d46f26f0479", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "e6e4e2b3e9d7dee6707dc047c47b2e15e818dc0d2fdd1298e8e5888ec6778175", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "5ecc061724007397472478391f97455c2bae69d8a7a923a164bc7d8795ba4263", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "b8566c8441f817ba9bac414495840fe96da0eeb303f7d138846cf160bfef946e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "64194f3950c961742b4356a0d3f9bc3548be56ecfaeed650b3021579f8bc6485", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "39571414f4ddccb8b094f6c8e94aa807041fec7623804803775482e68e3e3ca3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 768, "e3b2e29899b6d9539f0abd9cc8a4d080ee5fb5c767ebaf60059c6fed09fb042c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "964cffc3d9feed421b6432278ad747c5209d5fca7b2e985c560b709c5618e172", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "b132262908481fd9f47980bd7568ab484ad8aaa0dc2e6198ca138029cb979248", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "6041d91eb48da24abba347ed0e980958877a94d80c7112239f64203b21590209", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "0a06d88ac01f522ec6a2240a2e7cd93298cb22cfaa0b66854754123bd40e2180", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "9af10855685a7c8189971b0ea41cfd1f0a6d0071f5e1a1ac93aa69f1685d2f74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "3a4b4158c2effd1a074454bc63772ccaf8ed090ba43878f0c32d76e0cf008888", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "da130c23d8bb10b128e1e1810d7931a39984e0e0921bf90d7bb0f167fcff4010", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "b9efbfec982ba4c43498d843adc94582655b07cac97ef02c4b4563a75d889e05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "c85431edf29934fe4d0c31b0e5dd4b1b60135497d4159c0550a83bd5c7f94f16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "6f08405e6952eb49ca64259aecd99e8e9b543dc424bad9eb748c5906a5e49ed7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "4f42ba71efb430bf40ce10cf2953eaed10d6eefc9e783f07332c019884ec8d35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "b2215aedc7ec4e440780a05de3608c05cc689fd72b6af2c88752fb9321c615c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "b153476c5677f05d80ad056cbc1906fb56552d83a00bb49fe161dc1213e899f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "24a2a4f5caef2c3659983658331bf9c0b3d28f81bd1c2106b93d3e23bd64f6a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "0b7da1679b4a90d6300c5606e873efd43e851307e2c04ca9a6406f94eda3fd6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 768, "274da87584a63581f3662badff007a287961ff649e6069bf341fb11bb7e2276b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "a41954d286fbbce72137b557e8e3174c3293dd515ed7bd75125c0d846f712894", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "dd952231ce642e3a0674c72da49aab73f8679b1f784d48f97a73c14563334a87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "0d17fffb513fc7a29fc27158e06f904f655446dee2cd3edbedd92aa0fab65e3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 896, "e6105096427f36539a84af85e08ef2ebef6d11803e49982b316a0e55b7e44387", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 512, "3aa503d88b7c6de2f9f5b3af14f80ee88358a90c27db4bd3c10f7ab15d448c2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "e81335091edd1f01a3ca6e29d68b31ecfd98e3a040dadbf370f2f0f7f682d045", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "8b53bf5df853f8d3f4f6120a907eb67afc9a165bb4c08620ee1a8d88587e024e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "8cbfd44b09759ca2b6f31e5f24bef8b47bc13488c8003db5e964cba383ef9815", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "92f7aaf3691861a09aac0d09f9912c4c42cb35a19634c45511eed1c7f8d05711", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 768, "ead8591e18a74a00cfa073134a306fdd16c087dbc3bc05e3705e99df33b923c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "ec9aa748e00f0cdf4dbc47c3385d4ce110b91da27822f478269582d464741e5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 640, "5214d10de539e560114e058ed27ba5cff9290754e22988605436d3ea2c96efbd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "a6bd4b3b5fc926e92442eb2e1f64b15a24933aa2ae5b32a8ce2ac59f02a84297", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "e775fa77bcf7fca41352e5739a5c67794e3d1fe5f9953fd14c613bfdbe8a45f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 640, "692723962d261b03ea9e3b47bdedb9fb9f26f0d560863e041b68ea044c041215", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 896, "da47b0006fbb6ef0ac80cf28cbaff61d9b86191fa451de771a9335ed128bd805", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "5fa28fbb5b3cd5161130baf8512492b1fb90843c569da6be46fffc9accd693c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 640, "bc46944b89c7ec969f40e6664b93eada5096aef4bae19708ff886111dd675a3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 768, "da1f8e45cfad6fb2f5fa4822806759a63819053c28cc92130576bc7a90623cfb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "3a6b09ded929076fc218f69ae43ea04d5e9342435f3e2a88f60a69097f672866", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "88116e1717067e6c6421513fbe95d49b534abe59730ea44795fd5a28923d8571", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 512, "8083412ceaa48048f38c23bd7a7ee91604e2d836e4e5df9349c05b162e5071da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "49ff1c88ee294bd888efbacf65492b94bde73ffd691f3a5a7e6194cec61088a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 640, "af53133a6d01c1ab9f14649cde13df70318c3162706a1a211198293327ce7618", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 640, "e90126ce49c16779d4b3a9bb2296889a78d0819f5b8fb0cea06b327321ea590a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "0665ca236b905bfb681226ed35f274e47e075da8d8983b74774e5d027e87810d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "5ae5bf970c1b1a0cbe0e793d373a2d680eed595316ef433bc28e3ae28c3334ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "6133709757aba967960954e3acc6d64ae92ef6f0db4cd0b520dc6e0c3578a1e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "d3ca9201f18d376eadd5c807ccc64071bedc6ee9cc97819908dcfb8a9ee2d21d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "445015bab8c63f1c8ad402573aff66debe07a6fa87936065237fa03b40f22e3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 640, "4b3dfa2a701b30b017f49712f8d69a286055e180ee22fc0f524d973c873572a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 768, "5f3018fcfed7b430ca82ba407d6c41bce2d7a6ab4be1068eef90051c84efaf08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "1cea67542b035f68d5e4960220b417a62290e30566a56a5bd59dc3479977ba1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e9dc2c44d45b03dc3be9fa1723b3cd29277a71f6fc833793f222345048ce25e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "b9b4a219feeb2a5cd5cd3aedd6fcaacf2dd57b6f04627aec26fcd552e91f72c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "68ea4fdba3ef1f63d471ff78e6b4735d8d579a289c4227b8b29e308ab53e61d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0e9459e002cbd17774d4622bf395ed3db02f4f37e32f4bac2d6c277a70f709d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "166c634d80c0b85e86584d483d88a10104f899380952ff656d9d42ffb04c6bbf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d9858bef49ffd5e3ba6d1eead190eb27b1ba262c9f6e79e6b563ab5c00bae55e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "3538432cdf77081abdba66721e71b64793997669757338d0d5103c3c8d3da3e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "3cba6b80f97d71467880bac69a1879392d3bb00f5e4fb4a876461f7c7e284d32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "45df4b8d04ee66e8492e7a97b85548359377532584657f77ebf5923ac4ffcd24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "dd0da8f5535df5fe1a0cf4a2b560460b7089e989aec8f994923590275bcdbdc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "d5aeb25a6e1e7bc55309a2eddaee259024d31204174b55deb64e407fc263e0e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a7868092916e8b66abab44fe35b3e59248ac72b9a016a1be7e82a5aad71a3638", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223768, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "d5dbf53ba3adcdf1395294f837226d7b40a21ed33f691fce45e4358a6ce6e521", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "beedaf8f312ea6fe1bad0111068bc16c2e52b7f632f842c668e3517d0730b1e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 224152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "4408e4772f5d7aab6c7f2e838bd8bc901b952752c8b789e8775aa43f448423a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6b8f950efe3ef7003b4abc8570b5761da27651f7e1605e1f6bb5ed41f792b786", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "509bbf9d23ad6b9de1e5526a69c2b3377b5cdf563f7387963aecdf978822a9aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222352, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "854529ecb37570155bd4bb2a3d4b621414430dfa6ed6ef969847f64de364209d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "628cd5e25094c23fa6cb73f142574cdb514c93e207da72c313b8d2a2c41e90a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "40eac2770690e850149d5a73fd3860640ba13dcda3a17f5c5aee10bafe4cdeef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "61b09225239e4ee83ec8ca748827d91014a135b74bc8f7831d595967b9f3707c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223768, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "850e1b478faa4e706aeed186fc3bbdd137d4d67ee1c1816287995a595db6307e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "f32d7864962509a4c962b1685e362adf61e074a6cc678f4682c91a18d7e656ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183312, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "52f62196bcb62627a7f8631bb233bffe10151967eb32437c1b2cdc34d36378c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "42b130c6884fb141f6ffdafb17017655adb0ff54898a6bc801e5454bc9aca132", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222352, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "58b15ca7bff2f210754e5e8ab7699b51cbd291466f5aad241cd3a6e3649fd067", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 224152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "1be3ef80761c779dcf1631f97011a94a40499d696d8b1512a802828189736ae3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "2f0851f1fa6f1ea3fd8df6aa7d10be970cb710efcc70ae323c7c6580c2444fa1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183440, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e7fc71bb18fa079faef6600ea0862d7c4dec5588422e869a5a33b2e82537a2c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193552, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "68078a12e3595bed51517c6f1054aef20f77a371ab853912d253fedb13e01601", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203288, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "c4b63c46ef750b1b5301f245717aceba4acf152c442ef84ec35ba05cf3db0653", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202256, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "27577d6da2310ed1a6f43d95aaf10e841ea117c21019b02248e8c1e9c1badf2a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217624, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "ad96f578c43fb9a8ae1abfb7c48eec08374ef6ce98043eed2de7a96891e2dcc6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193680, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "f2702964e596d3639383b8cafdcf12104cedf8f9ecb12f3302848eec3603cda0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "a94d436803dcfd160bb73f4d80300f1d69bffc874bc94647ef0682edd9241289", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183312, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "384b3b50b7dfda65f0c4267f77a3be4644f67261f93e8aea0a3e4e84526e891f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193552, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "112412d250bde5fda525ec6c5baea08b4562e40437df1e9bc217ffd51135739f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216080, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "da0b3b9182c1befee54067c67dee2c280d2f97ff64ccb312c69be1fb28b4583b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202640, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "48d9b410da481a9834bc503e782000dfc7da011b26da8570e36f7d36662c264d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183440, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e95248676b85e2940891e3cf207667faba1c63bbd326d3263163da018bc011f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203288, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "93b560476f74972053b01c2676ba191a22b9a1bf91b09ae095a976b75c1f2474", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193680, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "73d829602a5f33c683b3c31e26f476381b9d8df88a04b82afed8a339a7b484ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216208, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "2fa8d1b66b175efbcc7844c1fb3a736a7668bdf0475419f759083151990385fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217624, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "7296d75840cfd7359802cd9aaca83168a33320b5df418d518f3dc21cd930b12a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "955a84f3cda321604eed41c7443571f5974b1c8f237b619ff0615321061ba295", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "aab307eb34b575e23ad7ca838f14266398eac87e9e7507fd0c539f34bd505d2e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216080, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "6f6516c9de166b5da6670ef11b5ad3c198ad56fc4ea91a8dfd4bba4733ad9bd7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a48649539b117a80269d2de6b9c9595e98b6b3f47818b15a3b36d8804e2b6374", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "d774cac6acd333b0b27f53adf5672ae818b8bd5fde63de2793aeb93e6bb06f4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216208, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "c6bdc61a6819c8541447acf3ded62356e18bf052c8b1888d4427af06a7eebddc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "326882cda63f2f426049ae204c68a71f55d4d196de133e7ca84095bd0343ac07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187408, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "28325e065af9bced8f7e40fa0d1324cb8f3f97ded7d4cf9371b722edaeceb328", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187536, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5509133d7b30f748e4549fbfbf38f6102928607122f6468d98d897c31779fda7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211480, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "7d8c737261f7241dafacbb149f020de66e5bffd570bfffc1f53bddbe41cd8a3f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 201744, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "0293b6b15285d10837c361ef814ee174f67356e5276a078494b62d881abc40eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "cd3d02429d9df2d74cd63799333634d32df1f2f807848e1fbf9c2b9d83ace618", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187408, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "4cc6e11181eaab1cfc0be8a3c90b962ef3b685fb2dd67682af5c8e24d10e34d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 201872, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "082cffa500237b1d5863e07ca713c040426ab70421faf3cba1ad969af39a1f28", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211864, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "9c3f3fb03ec0a2e6e0e8a52bedefb4a6a83bd77d1b2c0c39a8d35eeabad610c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "e86a951bd3023cc3def9c1f3f356badc048fdee9d59c01f0738c9d9398cad1d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "0f0a5d09cbcc2dac22aa04b9a512714044aa05117a3facc4c6f6b4d2ef01517d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187536, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "21b1889b368a7e47ab0fc8a8c75f201334d53115b2660c7627960003eb03fd37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191000, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "26dce8637eb9db6f603be94af9886ac1824534f9c49c33322e86dd053763976c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 201744, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "d5ddd332708c91c2764fb161396332e431f46de658857539dee6aa08b5016002", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211480, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "ac7bc6e63e349fe48d3e2d001b4fd3d444c67c7e03c3b2010a54df269ff298cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 208016, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "dab56ece7c954480e17da091376c126b12d95c8fb6772c05abc7506296b9824e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "d757a6e9fe015e4363e78206a5fe32ba98e77473eb7e6e9654f43d6d7b12c343", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 201872, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "e4115ae4c767f479b5887e1769a0c7bf555f18e80022de83e72639c212225454", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187408, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "c8006fb7dc5edf7dbaf6ea879681f7d466d5df490c4224b417925226a813a951", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191000, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "c3d647376ec21ef89462cca44dec714c8a340c1b67333d4acb376da9653cc9d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "ac214b4e16ee9646b0d399634235bb20272508e2695a12a6ff2007bc15221b3f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211864, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "c3f8d88585ff0e3853b272b126c1dfc2b8a119f7015a1da25491d12eae0cf01f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "13a77c91f3398ec5c2cb4d5e8642776efc6f67b2b7fbd226b797542f9fa4f8f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "289343ead7bfe17a830e9ed0897868aefada7280d8b93aba4d90dd9e93d61802", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187536, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "42c27063db72118e602fe346c62b8dfb827b0ac779689832942ecd26bb25bbe7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 201744, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "29b36884abf5b64c37823cbbab1d3e1055832a91a894884175cad59e09c73a35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211480, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "df151b0000248b81d4826ea52ef9dbada14d779a7a63331ca91abd8f439ef30c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "b9a255019cf99283b4acfc71c2a50a22d1b2bca63afaf0ddc02034d671ac2180", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 208016, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "c2ceb033652c30c40449d7d2c2ca10970de05a2df7639e1a31266b7345f39b1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211864, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "0acaf49b88d0b7038f73a2bbd18f9153c725bea4d41a5bae3ab736ea297e2f3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 201872, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "ea088add2e0efc7040f376ff2dc6f7f17c8186bda39bcfd1320d59982fdb75eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191000, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 768, "70832bc7a041b9cb67944590d4e72f9d779bbe63874470a056f5815b649a047d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "13f474227fd77bb1faa7061f1a0173cbe1048b5c92e328a141e7cf2f9df30bb0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "c84aa27e2e282ec9179ac16698ea49dd2a50200a81ef3c3b44b8f5d09f548d1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "d4c9e9f45ea5ccac7ef8dac96bca90fa3e29b96b0e5b4afec95332c6d246ed1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187408, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "29d50241dc5c0411bf8cd036702e3a05d47b71f9181357c292b4cd3fcce98e70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "5e1760028f08d46f6316fdf56221e0d28db5a62e807e23f13c90eaf5abe136fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187536, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "afb5ee64052ec400cdf7335c08551c66515d6fc6c8149ff8a2fc03fec7015f11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211480, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "b7f0ae6926e2054a2e8ff99e408de4b151f656aef0adb173cc27eaf4e1b49aad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 201744, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "a50d657e9e92944b9de6156eea554bb52dd77279585eff627b1fe45bdbf07ec3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 208016, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "a47916e7f64ce98f9b7e8db26cca79ceccaacb93e195c50085642fd1b7033e6f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188432, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "24fd9aab624bfef41606c341acbc3008e40d6e210ac9b3a6ac3bf5fd193a2f34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 201872, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "99ce9098eda55aa401d23364875169a64b9e1ed7ea46891416fbbe9bc47e01a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207376, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6f7deda62ce714bfb3d840c39b0819056dd1061ad549022760ee5d3ac92adb5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191000, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 768, "69eaa6b60f3361298299828b658d9c921a540a8c48e99d0e9ab651f4e37beff7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211864, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "afe14e1cf1518f0f07a89d287c2d49898bf766b8b87609c7ca78e4aa2f32c6bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "422c04cb6ce6ef052af4ec2ac9146d4521e9853e79484795c3914f52117b8c54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "9acb45d920e89c8feb1f4498e5fbeedab34235908aa27508f008d852cd38c732", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188560, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c8b2836e7024b337bc9946f0270a7a6a613648d61649b044bfca35797a49e413", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214552, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "8fd65b557d0fc6784aec1c1bcfcf7ec7d08f4d79360c12f50ff7642ba023847e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "3a867dddd31dab73088d89cfc73890362955e1fa26f671758f64bd23afc65638", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 208016, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "6809fd56767cac69ccf82ac026421a9fd7d411573ae7dd3c079166718c89c31c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "0a1a0e6358ca178a7e7377eee39bd112bd237eea2e7c12d78216278499b263cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207376, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "1a0e2d3cf64fd70f6106777be088d5986bb36f577f1e88fa2d953c2076dca2e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214936, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "be043381f433e635629a0ddf26689584f0baee95a4df85c68bcdc64a7a0598c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188432, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ecbc2702716614aaf860a8004e15ee839d16605f972904d0fbd5fc2662d851e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204944, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "ba9c83d6bad24790cf2c3d600f761fea6d821908272ce924a19da752adb12a66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "52039ec002cd678f529d7b5f20e2b56bb34ab8dd92955599157f3a77698db144", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "0568076a870fcd25b415513ac12f58d87174ea041d02e713bb1b16ce77079b82", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197144, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "16d9624d3e206539356d790fc3d1a0c569ba9a992df8221b27f9dbf433947450", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197272, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "76a3783bbc1f08e7fc1ffcbca51b629b0e676a9c70105934832622c28d273bc0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188560, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "9f778fce2c4320b5b59665cc537be7a08e9918ba92bfb3f00cf528befde16ea5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214160, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "8f34eb78791b190312333b12c9299f930fc9e823820f405ea47452be795c1328", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214552, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "15c00ab0f4fac28447b3d782325b6783f4ca3ce5b474389cf29e8cb18b6299ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214936, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "16ab3d3ae0ef32d8d78345f992471f85e6f9507db4372f242f54976db7dd2450", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "d53d9e3885d4d4a6a6947731bfbdfa30eb250e1f9a10d1a8b658c72934ed7588", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204944, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "fa7c8aefded7b2154c9a60f3c47ccf6873d456953edf568f2a74d306ee85c597", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197144, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "d5b94b2e72f1cd8cafa5982a4a6930a0e264ba023e5f4e14bedf75f6d8e43571", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "385a191e845c7bbd0c07b4ad685c5473fe199a8f8e9119c6ef62f64ac87f92b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197272, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "80088cfc247f0dd542f6c550c49b31090fc7dac1dbc4cc245c2e360ac70fd958", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "4994c3888e5d0d80749f6c2ed006011bb1d85dae51d65f94a6b4ad5cccd1b85c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "6f1460018bfddc9190164a39ef55ae8b599d0ffd3a9171b4e69cfe014a266541", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "8872a7795830978b6f1b1e6659567dd2d60e256fecc5083b45fe7f7b4ec98b21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214160, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "8c18058d29f42a141a13c7eb9cac29fa7fc35121820a1e0939d424eaafbdc3b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "2aaf10c621338f5555169e464efeec8967171b9ea82f5b92890b3b72a9bd61fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "7076b7cb06e0c5aa5f16136c2fb31e1fd409c730e169f0ab1350cdafb0b8c8ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "536a1479ca51d25e17601410c6f865f34690985ce2fb6307d59e736887660276", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "87237de8225db70bb3431dca9cebfb0ff496ec33c7f5261482d85019e6a320c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "f0bfdfed1f0284dab6b6addfdc59d9510babf359c4f175255317a907248d6c73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 768, "41fc1c2c287d14f29099d4c5d91c61dbd57ca52b8e2bbde63ad418e89699b3ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 768, "c7ef8a977688611b5672a5fad41724d263d2e24befc5e26cda89a07b1ad57a72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "893c1ff9a2cdd1cc63fe3d2bdc43621e55c30a77cbc28bca5f716ac02746e3d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 175768, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "b561c949280b6bb2499dece5e9f0a4f6dd3de68ae42dcfd58d072a58040ac169", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "ec75c558adbfdcf676f4e492d0c790b5e1f208cfb8cfa40eec31a5df54551a51", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "f9683ca23f12e9dfcffd533a2b99080c725e3a4ead0b38a20cd525de17e17bbb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 175768, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "491d414be11e103a7a49321869dcbcb12425f530c1bb56c9f494a38fdd12584b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "9ac707afc0cceddd03fde96d0c82681f3363276c8bd46b8184ce5425929c520e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "80c1c401962295f34ae577cf8c219efb191250fbe8010de42a4b1a086e861f47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "56a85c4d6530baa501a10c31b41e0c1afc9176c05242ba98b34eddff1ae4d806", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "210dae8432603496ce410c0bdad48790586a945855d9065881dc920b3dcdfdf7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "4a49a3f414568984f943d69cd6dcee779ed69af1e65cd9ba87a30c2ee57c83f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 175768, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "80e07682f94bb06e3c63836907fe80b22b566656fda726fc91cd43445285bc91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "57c214de0ddef83e07d8d10431c8852c754679de5c47ffeaa89753699676d9e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 768, "982b9fa6200474073bcd8c5efdb2f1434194d61a50b4808d4741256887bf8215", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 768, "e7bf6c84937d5543bd891fd90e3a5632453529cf356f1f7d79fa233fd5a01c8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 175768, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 896, "63629665e7d35587f68394ef2ebd23fb61c02d2a5372d0d470ed8aeac12e08b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "695be0138002e052d3b0a027924c45a30ee2a065bd49039fa3b8ba7b90567bca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "e5f97d6f286fa4344807d6eb1701d4ea10ff073d926687202007ceef7cfd2776", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "c61a024ebbcdbfb3042b176a18a0d3a4d024db387081a54162448b226daef549", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "2d67b20a99f1e0498c0d413573f3dac9729be3cd44df6b65a2bca0c0514edf1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 640, "f4d96105fdef2c09e6440829f66f082651e2617109caeafd23fef88f99b3b0ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 640, "6173991ce292b1c9d4d98e6e027c7ee7f26b9f08b34c9c86dd98b3247217e8cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 640, "c1546a5ffdb56ee51485cbd0a463308442598cb4be695576a9365fe2e803e246", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 175768, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "93204295ff4c2a36d57130e82570a8caf7d881efb630d3d06f9283a2ba058143", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 768, "77aa6a8112c9178d9043f0b9c9d3077cb1d2101c161341fa6162c39ee8bef65b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 768, "9238168e3bc8356a88ae3b7c77868371cab38e0ef8afce7cafb77da87805d52a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "1a62e6c8e18f465e05a05a3147f323beaef9804c1fe7a74ec919eb12682dcd23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "10a4c44cf37bec05a927a92149f8de3bf8e9acd54ee5b62467b0a52b87fe7ceb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "267fc1997335e45e0e00ea7fe877112c341ef23d23cb4fd47c1c2f7ff43bf02b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 175768, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "ceb146a677bbaeaa53cf3a352f616e420c6c2819f807e87c2bfa7d29ebdd7d6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "fbe07c6c3d565d13eee845834d9766aebc0835339578b4125b01fd827fee564d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "8d65b12808f4b4ae6b21d3932e9d791fdc38ed405467d551126df0cce9d4e4f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c6eed911f350b48a9d74bcaaec55760c4df60196857c20e1ea294c264ed6f437", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e8e09790b4f673185e040081201d0b2595d9e8f187f2d1e9d228616f83664630", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "d178c0682149ed9d7e50c5534014218b126f6194a11282fdec36078978fc668b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "b104f07fe1da34f9f115d692aba2c59f3855023210d20eeb27fff7534b8ec384", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "4bff07d52415c3ec4480601ac7ed623fd8c6a07945bc1e6a7c3f2b6811551e1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "899f701c799636952bf6ca90bf36725494b7cd7dd2cd29e0ee366007b3de0d5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "f8305530d2c185becfa3020d4be133ada1660d64c54073232244ef68f528a122", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176792, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d25ac109cb09d280a519619f53dab7951d43c4616d87a3e03a520cdb26ca2c9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "d5d37b2e66011b35cfd048a698dfa6e930447cae1b6c7e8f284abbf70920bc62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "e935910583f18af6d30e99a27f9d3200d751661f5c3396d99225640a3adace67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176792, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "ad7887d1811d562eee0e2d8ac0563c2e182def7b30f5baec16abaa6e46fa9bd4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 640, "34c6d4442c7ef290f8115dd13a121b44e8b837d16662f9f51ab969e3c3e2e49f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "16a1e06dabe6e29ffb5704b2e5d1ebc3e6dd169d06fefc8527341e3c3dcd1c60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 640, "bc644f02dfea8ac0b4e786ba5854741b881941252bf8553fd5c74344f6bb2c56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176792, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "6aae16fba8e815d8e20db4cbf22d5a70150e3628105ee6757e344c97c7ac2154", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 640, "b06f7bbdc2a05b17e6a3e8563915e27ab99cf44c6f4166d01ffe0fbdea7fda5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 768, "2e5144007b02cd4352547d57311f59594e5a04dfa93995022ee5d36cf2f7b8da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 768, "4b2b3059f825ad790ab585fa446314d053667532203559648fe16463cd839fa8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209816, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ea1465f0c425f36121e90807cac5837d8e4c592342b4d84f675974fdeb714852", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176792, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "e8b290e5de0a11572250810aba67c5cd4d1feec810809a37e3e0dbb0da36704f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6702329816d837b53a128bc2ba5b49501fbc699679f5abfebb52d80abd8cb807", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5f10ddce94786edb3fb5fc17ed2b5235f8434bc88faf51f85281b04a1a4aadbe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a7e463debdcbf6385bb9a453d99b4e253790fbbbf332da07672b5c462d802b06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d8ff490fbc54f39bf2934ecfe2e7998e1ae8a842d1c531b762b39c37215aff68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209816, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "fa613fba9bfc01908bc8af519c2a619538a121612059fc2699d38b81ea5827ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "30fb877de82eec0b8504594eb324ab111ecf2b1a7d4482d84c78a1eec8bc10a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "b9bc9ca94918a8ea6965ed93a440af7c7122471a12fbde5101018b604b857a8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176792, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d86fd20aed48c29585af41175dde5772fc5362e642e9458c0ec1d758064cc8ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "f5f89c73334ae6ddbcc7966c9acc44e191e7f318945eda42212c594bc1174d8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "df087af15acbc69b24371e351b1a3c319d411f7705a9d2d762375395ff442710", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "8ce03927720199cc7ece5934a4ed5b9f94f508aa5045329e51eff333ce01436c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "e618c10d5e8b6749fdee92f3d7b7a50d3cc1f6adef7852f2beb74d54db6f620f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222104, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "f0286af8e71c38e04ef337b845cf87d91a025caeaa46ce773076ff50915bd603", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176792, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "736d3e6623e09bb256280fc309209075f5601137d52217ac560e2613ac8c0099", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "f32a6c88bc7119edbb5f6add168f4ffd210cfe37518929642c8149ade176741c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "d38e28d38e4d451d3f16403b54bde96171696952f9dd3af02a1d2f998abd335f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222104, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "aab239fdff83e86217438d5e025ef262a194f326e0cd72496639d8a7408472b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "098d91e3fb8adf57500c4ddcb9824c177612071d62622f3a5b382f9ef2604b78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "f87a60f42251a35629ee92b631b5030afcd1fd0d569f68b0e697b2d38abe8582", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 768, "6714f06eeafc373d6cb9f2d022e714abdd4cdecbc3545b59987ac0daa98dd4d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222104, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "10f4c45bc84492889584e513b3b6e89d7eef92a129819d7798fdaafdaf7d1a1c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0d104775e4c9f7ebb65ed5599c8a29d92d4316221739c779d1def4f07e403523", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 768, "86712ae82bde295c9bb60f8fb656a0dd1cf468f37aae6264bf53668b48aae8ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f9e753d40b0ced8b4da12e2cdcdd0bdb58f2949e29ceeb468b5ad14a5c03cc89", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "78d5c2e98bb4606c3f1404728fe791501c9e18513947213e27788c93129f78df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "e96b79d24ba86c2fbc93da64301ec475926e0c37dc3b5cb4ee1a864956d2e466", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "7947d6a5f1bcd3dac05baff0cb18e5abc9d3dd73a7da910defeb5ae4cb2986e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "aae2c33174228e021641a3939debc1084281f3479b348ebc3de878673fbe96dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "629757e9f89b8a447325fdaea8ea71eba2e5b344301cc97a9de19f0ae85df6eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "8daa2c85d6b7ad04c37555b3747ad1df216cc8ddbd8a29ca56fc2bf1ce1c1990", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "20f00d4c44d50c05ce9058161f58c9e2592392cfb857808281ace39698345160", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222104, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "53ec067dea75583e3478c7e6f40daf44b731c26562242a0be09f4003f1799956", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "6b74e1bc3a4b68d94e5f281483bbf71db1f266b9f6cc9a4da17cea8f9a93f3aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 768, "d2a287b5f4de0012e8c6e46f59f80c488ef919979f3941a10f9664a15e8e130d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "b36c92c7e5eb650c5fb90b02f03e54439f5dfa518942141d00c7d53b1d4b372c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "f4fc20a2e3588936134229d6688cc04acaaa06b453755c28fdffeec4b0c5ccda", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "aa38fbe65c236895dbf6ef99d50ca528baf16d7c2cd497df7fd06048029afb66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "88d0e802080fd57a49fed02bad3dde3be07d5f0b28a5151c2925a3b2ffec49ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "ae41b377f8b08439c26878eb42890ba9d2f4aa705812adb2e52042c26811e81e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "b9d51f0a062ee4bd09f8f74381c4ab98399ee0f8e695102e9c80002e1417b48c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "f6dfaccb92e13c4c7e869b26811d3d174e966bdffe622240fb20ce791fb8e805", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "fba6fadc8abc86f114b5f6cf279432944ce7fda8a08f9cc31c31a5688cb16bc5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "aa6222cce02217aa4bbd08847283f1496c34d735c84ccdc8cd3a84c77c871b68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "23afad70c2e535690a48b6df0a8a3df482e27f2f2a2d4cc6b63f4f355b06134f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "e13bf8f0d8fb958526c18ade9cb68d8009d40307f1b654339b06a36328b32079", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "5be67f3b31ad5898cc12869342fdea6f01b382c321ccc3d6d076f65139cb7aba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "500018b3dd4bc02e04fbef89eb4a6a359ea05e7fa616c55c02fb49477cb10cec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "2d456bcb8f0a4798a4321ede7f6d55bac8db52bc2e9f99e33d1fc3ba775dc4ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 122520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x192x256u2_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "3e4a88f008bc824164007fd8997019222ae16b5f06cc1111c1e90ae635747264", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "41a6c8cbc2cff6ef594bec4ef1a0b96f7b908ce04ca983ec4613c8eca07a2e3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "51a2c32997518a5ac7393b27c56b2d4982607d601be6717f0a5d3925462719cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "8955f91c5871845b89cf3e7ffe05572652a9c207997dcad3495eb2967cec363f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "3ef063d74b9670a272001ddfced8dd2518a8d9d891aec39c5c61b9f814363b9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "a0de10e909116f5ce6591e159fcf682579bc02ecdc85c9852d1a30084b46afa1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "3a0b122d8a79de55a6461a29764d9862d6021547b33f485739bbb6409d851c52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 122520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x192x256_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "74aefbdc075b772e8f5a60d015d97cefddc6956d03ce5827559a5f5b9f37f2b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "bf84adcbdcef599f5587f4d698d4e1562280ce4f2ceefd8c175cc1aee5e1cf98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "d86be66228268b8e687c567d39d2270e99413c39a4c21c9193d06f878ad3fbb8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 122520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x192x256u2_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "51bbbe49c8edf6150c9fcbf4e06fb482a9915478c3061cb43e399c75632b37e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 122520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x192x256_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "10de053574f20a9325280d7b0235d50db69d421d9f664348969e0501f21cf544", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "4c68b744053675eda73f52c3ff37f550e3e0db06670b91eee2dd26d3d5c52df8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "8e8141a7776e4a2f654398e084f2272738c8015ae167dba408f0084f59df8c54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 896, "a51e8032ab87e28f468f6a3b400e469a5499fa014056f39e823468af402fb8cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "c7ae12c95c7f7a3687c930d8c4e3ad8396a0d0d319d2ebec772c3d92145b2544", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 896, "be837aae49c7e4783198d81b7580c3420c85cc0e9448ea223fb2a2cf6523eb42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 896, "c31678e27cf62790eec63abe7101ed8fb533ccbe8655d0313a17b02f0f770b78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 896, "dc0aff52769e020ea866788301622d4f209bf3f539010d0af177d4d691d03552", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 896, "bd973ee46006ee205b5e42d846f5d9c6df61f8a8be0f0cfd29114a0109c4ee8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 896, "0d32a727072be9311b8b6503464fcafd1d6ec8993608fcccd92a259397d4239e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 896, "fe2ece6202234279f31765cd3b301c428a2393f2ba74a2143e39b089d3428f9d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 896, "799b9540d47fef930de8ea970f1c6983a347b14a2e2f04f290f76413da9e4d3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 896, "a0080fb3efa7514ec56a4ec533984bc0c10dc349ff2c2b13841cda434cc4e93e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "9b0d0511a21d0fb92d616bb94c4a2a6c575f7f2a9ced58c0025dbaee60ac5d48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 896, "dce7addfe26d8d2c44928f8b3e37c4529f71ccbc83fa7e89b33fe535fc9e7513", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "ea4918aa1fa8a2b4b91a7ea754a3fd84809608644c0761682d933738bb0c34e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "54eb8b9ce460004dfdc699ade587882741f689eb0697a6bb817635b755621225", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "fcfb285c43e588c4113522e8133a2f079567a47ca01b36ac941d9bae7e7370b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 896, "ee12f7b0ae02e83f093acab335fda12af344eff70fa71149cee4911fbd4a9e20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "7594e5dd8e22754158ee21d092da47495d52c115908e6d3d2fec4e88868486a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 122520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x192x256u2_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 896, "3b40c5bb4c7a5060134ff018c7167203d0820588388e5796b22d85caa8a365aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "23883983a69176460a732d4fe640dbc6c7d83e6dd7f5ab98fc460b6f818e7391", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 896, "154e7e060a730d46a172d79ae41d048a6d146d4aacf1808a3dade0514b99610a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "6d0bbee61b9950e12a54759c0ee0b1e563cdb28a535ea891e1145875103bbc30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "2f679a525ac5bb6ad49259de2a80528b1a1ee2a9cbc5873853e849ad0725b78a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "1a72912a0b4e41a31b1b2399b1b7f1b9aca9db5d9e9759e89d110f04b9872b00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "a832f52f68711ae638c2c68edd992ec71653aa96d0d37d715a469b7fa07ab5f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 122520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x192x256_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm100f", 896, "02dbf4ee950158d915a1540101f66c7643e2bec9bca97f263eaed4bbae618792", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 122520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x192x256u2_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "fcb9c2f762a518c7e8bbdb7cce0263d8955ebecf1f8ac7b29281a76709ee39ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "9a1eb5dc212931cec952680e2f5d992712708052648bc2c4ae3c64630c7fb411", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 122520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x192x256_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "5ecb2111e34ebb295171a72b3ca01e8f9af4637082bac92f45eafb15fd2202e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "b36c705a667d36e217ba4280edaf59dbbc8dc852f59b2ec291698d9ca27276a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202392, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "f4660d01cf944d6b48a7e3a19a28386baed167cef2c19d735ec0498dfe21efba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 169880, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "fdc076eda35316ee873dfc0405f93e6a8043eff0ce62a1cfdaa8369641cc2668", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 199576, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "106c8c52e624a7ef66df4a9a6b0240a1e2ed830574cd11385f5378d14e16b680", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 179856, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "cf159411d59f542d5ea77cb7a605b27174c9d6e95931826e26311d7cd14d7408", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 165776, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "cf40b9c09d02a2416f2b562d94f13fa5b655b0b0e8582295192fcba6d9975918", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202392, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "4872708522d993a05d5358c82610bd12b15f4baf4c48c9136a114a7db2dc41da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135832, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x192x256u2_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "77e8bd9993ff898aaf9c62bee0b7c006934dc58974482cde66a071134e23598e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 169880, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "f2c01d7d5c614b03e385d89866b2554f48851bc60b28d0de750d17142eb95be3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209808, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "7ad85dc8a5fef83b53bccd97dcc3477e9740a7a9db7f29d23c4ff9ac31485473", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 179856, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "36453091c11ae84cc361183c644816c4ed64736a8f7ccbf67bf1f8fe3ab9bbc5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 165776, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "18cc8e62870a14e3ab7e7f1ef6a826383a0643b9b21e1c973f8cdc812a37e3d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 169880, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "64e8321ae62c157f89ca781b78873d20a83e5d35bd26e79831c796da5a13f1f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 199576, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "7239d3fd2399deae40fb7eeb06fe85ba44e88fc3e070cd760b29b2aaa9c365c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202392, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "71dfbd201546aa0c027c624146dee84ef69b70a58a68c9a4b6651b8d3bc60c19", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 165776, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "3c568f138cb3b1e241bc4028c343287daf318d07a4c75135e94bf4c11819f0d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209808, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "fdce7a9d55df2039b60a377d643d8501e96d4737b71f5fc1bbd7b48cdd8965ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 179856, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "b7397e727cea564ee3e130ada65d1dcba850bacac062472029c86dd5628cc875", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 169880, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "a17c380959c2c267a81880a9c5ddd970ce722265737977312ce2cf933daddfbe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135832, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x192x256_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "12b88b211754f8e1fcf924b15fbd693d39f6c3d94464e02bf9a6cceef4a70ac6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 199576, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "9d8e6a22a5072f8eb87814ed3cee8d92f4499f837a2ad4b324deb6b62fe39135", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 179856, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "de307158d68a1b997274156ec96929fb45559b0ed939a087c840ce46c6d0713b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209808, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "1ebee70ea912d57d97d19f50550a618dbb6c74fb87c84aae4481fec7da6cf7a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202392, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "4ad9d99d0574ae8ed820660ecd3643f435994157b82f549818ebc7400feeb7f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 165776, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "6e846aaab059fdba4ab335fb922b8f4154d23df1263de5dad3829f67c72993d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135832, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x192x256u2_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "87ef060ed74412b47c55d7e4edfe4992354e796b6e6a4ff865d0ceb71df21fdb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 199576, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "4c38e9d0940af863b47a602e15aa6952402cfa9224e475b828099b3aaa75d6d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209808, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "250fb1529a55750ccfb66899e25ac53781bccd65785e5e6536e3d2999cd11bfa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157592, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "6208ca9e1493c4d5204361cd48675cb402440a985609bcce7ec78d69ce802a60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196248, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "23ecda969c8d3679f171cf1c8f79d07c48e5013081bfcaa1d42f6488d18e3381", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135832, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x192x256_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 896, "2c5664e0aa58cfcb8192920dba024283ae1eef14af9fb50f585b9baec8adb3fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153488, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "598c56ec82d668c8ce6a7d95c086ec25f3f6cc15a52e37652bc3eccb5792bcc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173712, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "39c46dc45b9955dfcfec404738f92d5985f2e3114af880965d9d540fa7964986", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "44ca5a7f35ad8c4401d5e53e1d1f7e760125878f540572c2373a55a8efb3683b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "fc123687844de38031a74297c3a7370aa51076afcae0668b507cf2c0562c4530", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196248, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "dfc8a00c7eef1ae40fdcd7edefc825a60450762987456894c877094f7102a242", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157592, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "e70f1af19b1d029e902bf5a2a2bfa5d811a4af6e0589b62b718e295050871630", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173712, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "1377e56f7f30e22f760af3814edde97738b6f97dd7c13aed997f97261c55edff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153488, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "6df5785207d88606f6a6c1755abe48d273b6a17be3809a59c3d726c10541c47e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "2fce9aec4dc0ce0dacfabe72c2b0000481d8838209c79c8cab556237cecd22c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "38cf9bbbba3e1acb838662503a56dece8ed70ff43da2b7e189d354628a703735", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "d1450a2435c44ac3d7bd28b9ef4b57a20ab1eae2a973e1474f3ef13b006e221b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189336, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "46b79e74982d636a808d46d8af2a5147c072790b73f1c830d2a4c9174007c0ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205456, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "6e9a97661d419a726f726b90399b3debdc0593c9df3e17de5a6d9bd5efb5ab7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185232, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "e137c83a68f203265cca6a5cf1a520d53bf0fff9987764af5f323ea50b3967e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205456, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "e30ce9108e0441a0ac7406cd03c92c672083eae7e41d76db29709dd25f76e7aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185232, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "f84eb515100a1d6566f69bd0f0b582307866dd664311dfee0cdeabcf2f7442c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189336, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "76023c999a3e9609b2a00cce67db33b20babf672eb44ccb36c27bfc15d7ad77b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "4e856ec947aabdfc6fb180b9cac8dcd0ea111dd7a37d11666dbccc95887a49de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "5d13b480a02b967a77883b7f41b2b8d9b900c6c09471d7319b9eb7090f9ff506", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205456, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "dc27dfb18ec17846c809923eb3e02a45ade8a00805d8b4471c097f70f1d6037a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189336, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "f7d4e3b5b95b4ead6c9832d4d913783f03be10446cd85cf03dbe0b241f2fd3ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185232, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "c18030552abb55df6e7d5197f7bf6faa4e95898b0a7645bcc95e325aad35d433", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189336, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "c1f2fdfe1a2546d61d693b47fa2f2e4ae09674672a98bff99934bf66265a0072", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205456, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "1fa48418cf1422c8adbce4b0dead4e756c4a694939ac85f0c4eb32813e27ab85", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "4d9a2b423c47668846102d2f2f35feab9ff7d4fe2622bc17da3c4350e41cafe6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185232, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "dde6ebf772c9576f571caa503450140660140d265a8a6b2665ce449d5e7d6e3b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "eb64ee8ac9818e62308b8e7de7a99494618be30884df4478b74b4131a6740ef8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "404d9820f592601b601a929766672cb5cf92c9be894de909c36c64de6ea775ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188056, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "e8149f33835289e612ebe1ac75cf69b88a18dbe331b4f21f7732c15ec24cf58a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "9d0c2406b8635e5876e2728c7787d30459a162ed68c5bf5d7971ad1df2f37605", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "700c177a8d953df3225e3e5c03a5860710ddaf2b5b8693bc918a48810dab51d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "465c90ba173e58036d78db71ce69733be6b80c4c869ab630d8d2ba0de813ca23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202648, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "72e03b943ffd3570d3e8f38261900b21b32d8c70abd588dba5bf2c54aa8ead5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "f24e46a29cba9ed354cc1de12c4b57997a321a8c793f6184b3ba6f43fd0902bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "fc1d37e551216b377ca93407bbc98d4d5cb754306a7cd882cdeb0bb322f2757d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm100f", 896, "f7e137876a9ba09fb03cf229d6ba83b099f041c6e461e0514158f97decd2be8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 212624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "57e374a12ff128ce2cebee6b25c1d0dc808f6340553f5cc577c9c2e4842f725b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "c8ea1bf3d320ed08962d1e5856ad65ade19faa68145331a9af189f24f75113a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 198544, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "2c3e07de30de0b002eb7a8cb963c5c37f050caf3a594d5253604122de0dd8621", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188056, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "8e84f103056c721034f549620059e5641712d69481fa1dd927b81d7df1b60799", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202648, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "93c280b355bceb22308b0a094dbc06744054e35d12b161e1f62c92f6ad241dc4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 212624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "a9b6d266e8094168bd9101bad8332220bf96b7eee8a3ff098204cccd31d178d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 198544, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "ec2db6bb9b650b97d6222b2b82eb56f7228a345e67e7ddd037bd5294a0b8df58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "1a170268d1e9f51b3b9ac1f5a0542f459591f3d3e07360481632afc6e6181cb9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "3fffd5ede6348e76ba264927235092a148a5cf39b1dbd5e3a818d328b9a12bb0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "169c241acd078e0e07d57c516e30eff1a78181d874f41496a31576f00ad2ec21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "9d3c256c8585190e099fa45c4df9f62fb8c868fcbffb67daccdeebc2b3b568ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "498dca3b1ee34bd69f410c521ac7892c52e8e8413682a91a125a224a18773fcf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "c5d16ae722b5be7398710b0de1565756c8509c8f1e0924550c0565cebe3ef68c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197264, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "4ac6b7e2166a87b187f0cec30b379b330121636afc5e7169f42353f5fb45f7ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "1a679723b1fe7b081acb47b36b274adcc359adf274174a93499279157cbe6406", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "ce2f5368982db71bebdbde7dd23aff239bfb257c869d4570f2bd0215b448bca7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm100f", 384, "5f8bd19dde93d4fa204be61e8649c53f84e343cdec3c9c8a9b0dfd22f0f015b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm100f", 384, "b8fbe9c6af373d76e5664260db15178d0b1420d9dadf0475c90c83579fd7fecd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm100f", 384, "b6d98f97500a4f70906e79fbc57b284d5a57e22505b2609d8f42659b03cbd3b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "e11e19ceda39d3f3e0365275a646302aece96600a8c252c39ea96f336ed21466", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "bfec0574d0f7da990a5514a13353729d85b80140b8613fb3fc69d326e4d326ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm100f", 384, "9ae8f69b20fc99b1edce9d06e5b470c20e48c5b0ca8d8c6d09ba43fcbd078bac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "037e13b375f354ea367903a313520242e9ba03ba8bf238580f19ddfbffccfd2f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197264, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 896, "6fbd02f8ba3c22ce6ec6d9963c0361689121b8f41d11f160b59ea067bb858f58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "53a54bc0f7146504c4c950133b606ffd69d5c2dedf717deaee9e48e33e3e0880", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "c7f19100a1e8edb49447e98e3f7b8d6601e7913eb045fecf9962deb74c5d22bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162448, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "915a3d02d9b92a3f12669b3c87087faa221e9a09224103b8f6db51ea0f322be2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162448, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "8bba5825c285041224d904588d315375c8320ed5acff1f97a98dcb0155ca4909", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "1356904eea2d0d07f51d3db9bb8ef40e2f5770807eb96bcda62896f8586c522c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "cad6ade9ebf417987c7df9e6637a35d2c86908423f539aec20e75f4157679368", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "cf1cadd17f2d2aea5c19336603952a247756e3fef8ca1ddc3d074f71ac46a8a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "bb9b2797c604598ea6cad33a7e1eb69364e7a71568fa1cc272188644a45a4920", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "65eca5b759d5d9619b80df7e3359c79f4969d06c9b3117a0340ce0fa65846cff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "1034e248cf131d46fb009cb8cd6340523c0ef6a7de16de62aa8a08a3e144d41a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "68d20c7f0206b48e24336cc6c21312468298e15ac42379c612b9c9bf142fbbb9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "a8751784e3f9e78d2c55b8c7b79384d0c688b9acca80a1d3eb5a40b612741ad7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "bd233835e5a60cb5ba18a298664c3c5e37fdd6f911d06f2aa1edee19c4827685", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "1566e23374a3267828d835b527e2b4dfcf0cf59a58f7d58d38fc8b99c6cd65f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "9bdbf452ff9655d9d09dd1ecd599a702379162f5e2febe8be6049886e94a0732", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "4c84f725fcf75cb1ea312a31d8366b81aff8f88f28deedbcfeca23761b945f52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "731b4cf783c3dd14d139db258a41876d6d4c930377921129bf07265d1552dd72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "ea532cf82f8555d92ba21ac2c0b38a1b8afdefc226165945da9d57ef4621b680", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_tma_ldgSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "f75841e24dcefb926c658ed23cffe359a12ff7eb9f86990ae7e1079f17f30fc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm100f", 384, "942a28e3dd7942db8f51f542bbafa6153c294dc5ba0a38ddda632669880b8e18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm100f", 384, "35685638562db9fad017cd73be1e24fbb63a24659dbc7fe95f0334b108351d0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm100f", 384, "8f1ceeccd7dc6c6318fbc7a3382026278103abcb9545b96d3b25731368572dc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_tma_ldgSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "ffa306c25ed33dab2e189afae44a9bff6a81cc0fcd51f3b86f8abb7e362e5b75", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm100f", 384, "2f769b987e92739650ce4922b8ce1872d11050fccb428061bd0541b4a6023f49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_tma_ldgSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "cca36d3f4909c81b58414139e87b57beb1c7bd94f5f4044275f8e01609fa6c30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_tma_ldgSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "8cbf36df4c15abb995716ddd099a3f9d99da3d1b261c3720f0b475d2c4ff875c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162448, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "5d9b874686ee43176c0d25bd98bfc85572620e62726361adaab9653f313c3fd3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162448, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "916dddef3b42c254aebfa465a230187449d0d36f647101987a1d3dd3699c4b58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "e3a500c41ad302ecbd6bbb332ec62b5a9bf9ed99f56d43dc58186b5229734bb8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "ddd3ef30546eaff8ea0e72b27f495d8079e6637fa1d9d1d69b6ea95ae5446222", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "ea4138da44724b374467db66cae01f8adabb3f8e287c1f7b0fccfa5fa6a81a6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156304, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "28cc9be7474bc733a77911594ec0b5fa82c4a887139252112ec5173bf27bd1ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "9b6ad8c64f8ba5032a2e8764c5e82110914640a6ad98936d588c4934cf463643", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "770113ea501f452b102abd7e6b2f66474246034f6d72f21d4acbe8efc1ff775a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "07845893e45ca3ba9dc57bf7c705499676a6f5ca9d7779e2e3dde530c7108056", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "7eef91fba84baa59deff8bf1611b432d40142db38e856841dae6a2b071a6f602", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "5b39616effc6a1c20200f395bd8155b3bf9b98eb240a61afe7a34db0d3163c07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "90390cba77de10d03d173f815f8ca6b2d857440950bdf7d112d3ff64e839dddd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "a9fd4c46e93836deba1301cf06b8200853d74534e024e7b6220b0b95d68eccd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "5e7e9ba6b6e0c134e3903a5e56fdfe5819bf9ba0b3e08f20c246f0524ddb869b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "db1e796932404edee16ac38ac8021740af2e9383cefa7876b607569191f62246", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "5f4596a4ce1a7c52a09a2d1869ecfd06f50711644b3b2c4ca4c6f5bd020dcf05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "4bc1f6d6c97a2067c2242fed32c0aabb74d7783e6e6d9a2fc3f8dbe5a61b14b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "90684f4576cb47cfe0ba516fa57dc04eacb376aabeff8a1555c2c9e9a2f3b524", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "04aec64baaa01838ff00963a079ea5b28bc80420fc8ff54a21d33e72a311549d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172696, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "6c367190f89d9afbb21062e80856bdb14443f26f7e43d7d9163a5f23a14ac377", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "14e64ad53b81cd6d90daa1d78f8ad06f62712c7b3d0976ea554ceaaa34bfa0af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 182928, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "2157d4590712c2be7a5992a369e57418d15ca688d3a42dad492e4e86048c4d93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172696, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 640, "1e3fa9125f0e73fe89852049374a8efa66011845098a34b402214185a6c303d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 182928, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 640, "502c89faa080fbeb7cfb50f3a1868f8dcb64ba1a8f08973c3163447eff446211", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "9658002dc41ac850d4fdda662284e9031d432e898821902f41e902edb353b79e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172696, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "e39023d1bce6a1f1984bfb22310a2ddfd13fcd66bdc8664d92aacd8635e2e479", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "47ac6b6463ccb5f427e0ab15daa0c098a895266e09be7a4b2437a61428075d52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "3e99a91d4bdbc8c9553a9aefb499e4e051dad9e2c899e1c9e9142d7e7c1beee8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "a9bf8c96300aff1117c0bc0b324560aa010236aba22a924cd4f9298a07748299", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 182928, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "d83bfae0177cf1d94ef3f7fd5e8c0ab51c2f4c73e0d20719925bfd26e5e54397", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172696, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 640, "439d4035bdb432bb3df80f63b1741cfb5c3b00c5c79f3615a60390090da96d52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "483f5ac75ad759ed8cab174c015c7144b749b6dd30a68d41f71d4a47f24ad231", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "ef6f5eee7508d6d7219bda8edcdbe3ab44fb6c45fe172b66625f0d02fe9f56eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "0146094ec55c7b1bb8c43ffb3d7262024059350e83fb2f13ef54fbfad393af87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "2fa7df67cafbb5018da6df690f88c3d1da1a25a94f1bc45dd9a0a31d2842abc0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 182928, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 640, "05c335268fef6606688410fa8653e1ad8e1eef4471702b9fb8241d4454030fe5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "e48826add76c69bb2ae4236d8450e653b009ad61f247b7129e04c75f70d72ca8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "046955faf47e0171d771f667604903fb3f9a7e71a36bfe9477da60e6d134e393", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "a6d2b50b6ed2d1105fb62bccd33301aec1ba36f63784c1aa269ef434567efb24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "648097f2175f8fd32003f65acfe88038b91705ac809d0e67f15ac1836945c278", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 198296, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "8058826edde1f1d7a6b73ae4dd8a8aa98c71f51296396755c015fc9f45fc55a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 208528, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "f40d356e2565fbe99f5ef3c14f1d3850400d411b7dfdc45a116da0f9594ead3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 198296, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "016f0f1b825b8f1a33e71c6056b49a0f5b5a588e0a868f29686bae15085a9ef4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 208528, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "e9105ef32648dba85c133340caa0aa6495c714328701c5665a4c09ef28561e86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173720, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "7105f802fea870f5a73085fe8d6fb7a378bfb3d4cdab997ed503bea7392a40ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183952, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "33961d82bb5cf1eda56f4d4456ddcd9453988f44f1f029f566c6d5eadc5b5ad1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "f3458c715e7f78742ba66f781504abfb99bce6640c778752f798e2d07517f54d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "bfbd65d6a43d00143ec2ab63c79d15b740a87e339edf8bfc860142c3525bfe63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 124688, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "65f86fecb9efeb102de3cf553920b8e8a63d23e009f6c86840e473946752d3d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 124688, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "3b373c155ec5b8f328aa05630b15462ed42c4c475d786a978ec84b3bfe80ccd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "037adf3a5ae1928dd6c6bd71d95d5dfec9c0b4d66c3d88cfa19e08c857f0483c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "70af6cabc1177324a178589555f618b76df6160b91963b0c5c43f017ff3adbd3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "82fee1eac2e8b383bb4abfd87c9a11d8434eca152a96f2e23e71388beb452274", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b6309685e6a936337a2adb0dc3dd7898692a759438d08946ab884ba72f50081c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 124816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "4ebcb5e9ac17f8a895b1086f949a609c2439c744e417ff71b73688f659cedb32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "67343424cfd67cb84e8b342019a3a1d4e5016bb0ae2d25ad93cc6be5440e207c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 124816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "d06a302b639efed42706230586f388b786014243cc01a77352f39ff4cb76b64b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "0908a3b6a0bcfe0f36e91b9bdecdb4e99e05f5e55efa9a6d4c315fe92b22ec10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185752, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s1_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "fb771291e3298afefc8b9f58106bf93d80370a984b9e5fa46986b7708ceabed0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s5x5x5x5x5x1_et128x64_m256x256x96_c2x1x1_rN_TN_schPdx3_relu2_fCp_tmOv_bM_ldgsts_ldgstsSf_rgTma_clmp_laW4_lsfaW4_dynB_sm100f", 512, "b31a7467f0f9baba7dc1e111106d10613f91a1e9b8b36cb051a8a0a0d6cfd042", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 120
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(0)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 4
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 4
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 169616, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x3x1_et128x64_m128x256x96_c1x1x1_rN_TN_schPdx3_relu2_fCp_tmOv_bM_ldgsts_ldgstsSf_rgTma_clmp_laW4_lsfaW4_dynB_sm100f", 512, "0009800620414256ec125800cb034bfe0bd576d5242e6c690ca3b8face1a9e02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 120
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(0)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 4
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 4
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s5x5x5x5x5x1_et128x64_m256x256x96_c2x1x1_rN_TN_schPdx3_fCp_tmOv_bM_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_laW4_lsfaW4_dynB_sm100f", 512, "66e43c89f2ab32cc1c869cfb94ab13d5d89cc99be2dce558c4656d625f3a816d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 120
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(0)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 4
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 4
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 169616, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x3x1_et128x64_m128x256x96_c1x1x1_rN_TN_schPdx3_fCp_tmOv_bM_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_laW4_lsfaW4_dynB_sm100f", 512, "613e69597cdf9a6d872079bcbdb0209eb9949abaf1ef7aec9917567aa6f81047", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 120
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(0)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 4
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 4
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s5x5x5x5x5x1_et128x64_m256x256x96_c2x1x1_rN_TN_schPdx3_fCp_tmOv_bM_ldgsts_ldgstsSf_rgTma_clmp_geGlu_laW4_lsfaW4_dynB_sm100f", 512, "04f1d7e1e99f29fe657835fc1526e0d94595b5dc36b185397578c06ffd4cfbb2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 120
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(0)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 4
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 4
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 169616, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x3x1_et128x64_m128x256x96_c1x1x1_rN_TN_schPdx3_fCp_tmOv_bM_ldgsts_ldgstsSf_rgTma_clmp_geGlu_laW4_lsfaW4_dynB_sm100f", 512, "d953dd72fa1276de9870640554bba52a0803b38342a02665e5d7b52b6807eb4f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 120
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(0)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 4
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 4
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s1_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "f711e48fc886868c2c5a1cf5b00fb25a9265266aae32c8c9573b8521dd1c2244", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s5x5x5x5x5x1_et128x64_m256x256x96_c2x1x1_rN_TN_schPdx3_fCp_tmOv_bM_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_laW4_lsfaW4_dynB_sm100f", 512, "eb4d45c9f4ae19c7f1084f9959995638b891e0d9987db5074cc4c1b608bdcef2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 120
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(0)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 4
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 4
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185752, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s1_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 640, "d6d5d0c899ff77501d7688caa268bcb62f14e29c002aa6770215d0372c05da79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211352, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s1_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "b258f25e8e7c9472aedd765e5495c6198dd13710d4dcd7a60587fb1cde1717a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 169616, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x3x1_et128x64_m128x256x96_c1x1x1_rN_TN_schPdx3_fCp_tmOv_bM_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_laW4_lsfaW4_dynB_sm100f", 512, "d1ddc46728888a4e1262f3aaf558005f5b5ab2a2c54289ca1f40e2f90a99fc5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 120
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(0)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 4
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 4
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 202128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s1_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 640, "69b7d00a2f832b9a6ce1d20f724a2e7f44bd76cc6991799049956831422641e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227728, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s1_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "6dd410ee563d9ec8a858841e662a38681078c267a3b011d0eb51336cd8db695d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "2440fe67edbb005b87d15cd4f4012699ce565119f7d041501df7f7aae6eed7f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "985a5922ddbc4f8649e251197566b185f9ad2b2f83ac7c3844c7f446a7f4c307", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a3b816b538d7be91b76238b7437b477fbfe8ea0ab3ee3c71e0a48c1cdb48fb1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5a354c686b27344f6724777fa75f83001bbeb9af3377e2500a019623703c8044", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "da2477937513265fadc986db545fe4d5a00b195e8796319ea95345f03d65caf8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "61e0b25d5c57218482e8d2d3b5253b070c1aa7cdf789c81188294316c0ae660d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133904, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "c26d58f92ffc21ccf081f3ff0c334224b494aefae1d9c6b23f1e6c35625046d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133904, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256u2_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5d84d20d74191ef794c8d473eaa92544c5727eeec17dc97e533ed2deae1e5454", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "220cbfd2f1da6b39da38a8c825f94a6fb85c08652146b61bd7801de6eb19323e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "37820ba506be13e505455317c14528a7a8a5bf8ec149a13715246f144e2437ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "fc5253bd1d913f290f0024e5e5ddbb600b1304d05dbb54ddcfb7bd2b95a7ddf5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "c8a0b5ccc8c143d3a12380abbf3b2b23e29c503089efc27213fb301a56e40c58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "1f5597484987a1ec9d645a316ad6f366e811fa8d874719e2c9e0aa0fd73c778b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256u2_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "23828bf60fa4ba7d7198ac545e811df92075be8275baf1b5e7f63eb0e4ccdf1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b9221e98de9acc1f88bfd47dfa4611d86c70af90621d9b9e3aa91d536a053e04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "e2a8300105de053bc64ddc75ed3108e5d5c4ab6588372bfe080a86d95026e0b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 181784, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "820951d7516d7c565a7f0a20de2491ca14245de2043fbb3dc7a83f13728abc74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "4bc6dc7be21bac82478dc52c48124150205178ab8a50a5575082a04c0daa4003", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223768, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "814f64032b1840e42366af990805383574ca1d9fcd26575b60fb9a8b48cee592", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "76ffb3a31c43482d7138165daa7e3017d7bcae8567aaa826ef053af6390d05b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "d220eb1bb93f893fe769ed38794c70483e8bfef2154e2e466adb18732aff9d08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151312, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256u2_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ec74fa9b105d081f5f265b98ca05e3594979fc3bbff46b8a05eba30aec856230", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 181912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "bc8716c5647fbfd9f0989e328a93efd52fdc33ce709b33b4d7c35640a77918d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "f396fbedbd0747f4daa221cdb870c3d60512e61ca507207443bf4b3651c21c6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221200, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "2c29789cef1d948814b4f75908c371ebf5021308aff127f07cd4660f789c1695", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "77e110563a1ea4d8f41b2e729ba3b3ae5afa5e13d8773115e2283d0d550ed50a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 224152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7d2df63d04a83be6b58fa04e47cd9813116fb5beaaecc9fa9a595fcd672780e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "603918e812ce255712487cf76bcfd10b87e34717d6af32545871206dd9743d78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "4bb5f15d532ab8e7e3ffb10d732af234e2266addca812b103ec9aefa022d174a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151440, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256u2_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "abb99128b910c518e7464b68b65def9c7dffeb5b72df9815703800b1cdf1f276", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222352, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "3fc747ed8a6f4444336ed44606bb664587268e95a0a37fd09bf13697b0a08f7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221328, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6af2920b2602dd3aee25a182bed7994979900bc566bd55ad3005d54d28c3a828", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187920, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "267dbdb2169a5f5506c844187aa2e452c1f5f2d49b9f0b61f7f1cbd77648cb7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187920, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512u2_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5f04ad4b8f687644f24f41555fbcf31c0a99ee13c8893ece3e55b7c7e792d5bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188048, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512u2_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "626f66f9ae5ed132d00af3338288cb987f4d73f36a7f846d3510880e2a3d98d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188048, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "d464380448042f455f7bf51052fb114e5aea46859751d686afc151d894cd064b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "96a9906f08d278f8c8571e412361d75364224d3705aedc093e1013e3829426a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6c0fd357767fe537654d518fd2960a758713b8d7818d2db21e50eea9e0bf38bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 124688, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "1211ce5f7816034546c4aa395a924416fad86e44f526f4611ea235d89261d459", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 124688, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "c7f6b71ee0fc822aaa9c8cb5f6662694d514aa5bcf8b8f0f39d78cd05dbf7df1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b4767dad827451b04c8ff4f9fb747341c1e9d8d09b7fa26b5200b96013452858", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "f3acc27c4c39e2b6232941c91a3c4602ec9081aef4a5f020094a16ca23bcb62e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "bd2918d4f43dd779892309494c659a1c4ab17d91e28bf79041c40f4c9afbf98c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 124816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "e6359caf59e1a0f40dccadaffdd43ce6492472420b6c1879c0947f2d4d1623c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "4772243eeaa65e8da92377a81480b29adba03e6e312a379d70f45619f1e5cf25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 124816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "d62e45643d0650b44cedf59f8e0ab023f73c23f02ff5c362ab9150c10280da25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "df40bea2c8890e78af8a5b111b5ec419c9ff25dd5001369f1326448c5a424a5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7d1b45cc188c55ce4b05ecb506c150c89505d699940f885068fb193190f8174e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "f9533385050e750129540d8bd2b96c6e9a75a7d0b46fbb49571c0e8021cc05ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5dd493a7260ee17b1ee264c1bf83c52269b1df75393ae337062137129a647296", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a960189b29dee5074737a1422a1447d36fff85f3bc92dc746cd550c854585f7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "03958fa294af2e0c5db1cfc00c5c248d509143a5254b4d9ad929c889074c63dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "1ad9837c66175294f0133cc4b3c3b4f24f6cff8ec9cb0cc7c90157e24a9f603b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "12f80e3f4f52bfb0dbfe84670f39401c8d26b13f34de3f1747538ec4e74802ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133904, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "3faf72ef6ef2960ead9836326832f69f0d9132649f95d610f85118cee39e18ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133904, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7641175540fd5e5f526d1f359409cf8b77b71a2ff3d5daad63eaeb7898d25fbf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a9319c8a125de48bdd33c556b8903013be62ca8c4f9d55945c0d58920b625451", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "00069c09b756dfc542a6602832f983f31760b71c2fa7400e43b4f10437b760b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7e76f0eb0d1d65ca3c84e807457be65060122da970345e616c9460cc390d797c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "bad268baeafd6954f8402059cbc2cbe7fd85fddab8e848d072f9bd70b3651e20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "c59f1bd11c2579084560b9f1784ce2eef47018f4a65c54a75ffdf7c5387f659b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5be3590b3a0c4eb3b4e9867d867d4cb398adcd6bb604ab80ac7a59c95bfd8c86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7d079b4df6b64974d6179639c76108c56cf4dc2289543077c376223eee7a5547", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "d35c2e79a377efa9439676a0f182a49750ecf03687f59b6cfa7cb0c00c558efd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 181784, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b6611819761f10f3a9a3de6936c7584a568df2e537babcf9c58e20170fec8edd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "1b939da69762e3cb445642e2f238832b71e0b8c702346c5a2b036dac343867ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 181912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "3220bfa53ea7bdae237a608fbee91c88706a725ae047ec2bbe6e4258408f473a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180888, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "4054bd6b5112473a7ab2c3657a0b64d1b973eb536baa13d5933417bb9ce7560c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223768, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ef06ad366e7ab9dcdbc9ca9e1478dcdf188c42510e9697ceb68ae22d57d56a61", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "51d8344cdeab1aa4cacc1b91b0670f283c2de196be4f1888d327e0fa7386faea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "f6f0c5eaccd488e8ca248e04b2b25c2f0ac055f6906cca5df576d056dabe040e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151312, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6ded89f3a2d70fbef71e4f15bb4cc1b952045c50ac7ce1cafabf83f575ff96f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "07ef37057552f7dd90d589fc345edf1492ece49bcdd947946c169fd475786160", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221200, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "807f85df6d2c1fb287d9e37483fce6505cc1e22d73277471f1d0d3c97ae58429", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "3d3f7eb57e2de3c610e1e8f2b1209fb261146a4772895ee85a082af4a3456150", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 224152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "42e4cb4008e3d2548ea5c33f30ecd9eb3492b3c7292784ccb18decbebf7a3b7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "526ce0821758cc1c9719f2e9d849db6244f629184df869ff5b7d970a7d07da15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151440, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "003b3c4dc35fee3580f88c01aff7fa968480d56129ba245dfcc8b2f3292505ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221328, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "25308a5ce615f49a10f11f778f6ce4a32459f36368acbac57694fcf733b2959d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222352, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5c565513ed479c2df059102999aee925f0c938937c1018e36bf6166c4a29018b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187920, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "c5e45c3bf3246d11e70a45e102d04462c0f04d3fedeba401f9dde20192d11e06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187920, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "805229f0b88fd659425004dc972f4b87cd1c645b1a1ed52c7984aefc5e8812f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188048, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "cac4c31a5746969f3e78a9f55f38a734d43d032901b746b729dbaaacbd6f8b81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188048, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "100bcdb0a56704985f79aa1eb81191d369f6af00093de76e94f3378a94b1641f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5f5f23fa51761cebede2b95b0effa1b5c44d29165a412c219c8f779fd65e5101", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 209816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "30d4c9e52b0c12af5a7d105fd0b1a83bc3adac9b024562b8c9239d80a730597f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a65c0c66349a005a85de81f010856b5a2566b805a257b1d2521af9cb8343f671", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6f9bf23ebe85211a61fcd250d921105b131cc377ea8d4ed31f7d4e6da24fb0c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "8e3935c68e80e733f8335b2173a49169d7f268e31081a01df0a6de2cc8a1c172", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "35ad47e62a359613b08082d1f1f28072cd28a5ce7228e5a6d46379af17d7bb79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 210584, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s4_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "23a73d612f4bb85de16be48d8a7c687d1c1ca215608bba5783a75c280d4d26c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 210584, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512u2_s4_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "9339f3847af1cf7604bb1a2adbc1431afe6925f6bb0918aed2afc92851ae3cf4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "ad2a3f54222921a55bd643ed8fc7e8d99f793481920a93fb566e0943d24fd304", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x128x256u2_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "af47fe86978f2b6fa444bdf47f70f25c540389126f042277a31f648a742b0ef1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187024, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x128x256u2_s4_et128x128_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "ca99641539785924c729c5872d9f43ea3521ce239491ca7420783ec7af95e112", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187024, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s4_et128x128_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "2cc478a0a0a9fd9e45dba79cfbb9dce7fa47eec29faa018533b0741379dd51eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163472, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x192x256u2_s3_et128x64_m128x192x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "0885796ae803ece586449ad8a4869876477e3cc6825026280c9ba18343c14da7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162448, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x192x256u2_s3_et128x64_m128x192x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "cdadc96b414bfe9db2cabc28fee3ff7420968681ec1e0f3d0da7c471ee37b319", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 210584, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512_s4_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b18b0ea30ddf24125fae788dcb9c02edaf87fa25eb203c993e5ee0863556e2f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 210584, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s4_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "fc8bd2ad5a0dc22e654e48bce6300eb71e4b2ba1ed32ad75905bb28b3cff7846", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "6f5d2df2cd68bc8a5bf56ec7cd1f14c1a387191e9739e2e34fa9da49ab4b063b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x128x256_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "a53463ede308234ad1befb3d933adbf8e4af06a65367225d98c76badb1883770", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187024, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x128x256_s4_et128x128_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "063c926a934ff59be46d05edddafcac2b5ebd800a7864f2b7872de6cd3b71c09", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187024, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s4_et128x128_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "ebcb28bc89f5efaf8f623b8f89f409c36e2a06db2a78d56b95338510a6835315", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "442dff92e3cda8ffd8fb3da4247c84a5fb36930bfe46e47f8b81c8e44f1ae700", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "d56c62e564c98e3d4538202497ecefea27ec52209fbaa6f9117fc5161242e6ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162448, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x192x256_s3_et128x64_m128x192x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "9fd54819af9837dd897fa82d6a9f27b1305c60ae725dbe9072ec50c589acf150", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163472, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x192x256_s3_et128x64_m128x192x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "68a4c965702a0f3aebfc3b4222ff04a285fec7d6853d0d82bd95596683e4b3a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "cf80d2b4bded665ea33298e227b49cb778b610b1de734909b80242e7a2d0651d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "db47cccdf0afe082fa31f5f07411e5dde72ada017a5ddbd6e7d6f7a87d50ed53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 219800, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x256x256_s5x5x5x5x1x1_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm100f", 384, "1da013447ce3fd14a4810d5203f4213f5c07c1b846cfc8b8317a45df32ec85df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191120, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x256x256_s3x3x3x3x1x1_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm100f", 384, "747c24c300fbab9c7ed2b7e5583f3f9f546effa1faea153426c036321e89ad7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 181904, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x256x256u2_s3x3x3x3x1x1_et128x64_m128x256x96_c1x1x1_rN_TN_schPdx3_fCp_tmOv_bM_rgTma_clmp_dynB_sm100f", 384, "f8c7b1406f5b280fc3260d6afcf2f2611ddf2ea3035082d139926780b355cae5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207512, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x256x256u2_s5x5x5x5x1x1_et128x64_m256x256x96_c2x1x1_rN_TN_schPdx3_fCp_tmOv_bM_rgTma_clmp_dynB_sm100f", 384, "4071272d6a5de618f528171bdf08534c88c404d0bc10a343caee5224221120de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220824, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s5x5x5x5x1x1_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm100f", 384, "d8c763686b372e98a06d7c2f5b2203902c977f83577f642753d5b2e02c7f5f16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "1538035e2c6072c4f961702e13eec4e7a692498047d7c7a23e513fa964dfec31", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "15d926c39e88ccb44a5377e982301fb6103bd2c71853dd20fe2588dcf6794dbf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "537c81e75965c84c68d2d194a4738649df2e5d237deb7f9b2352037b63b0a464", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "b287ac9fdd275db13acb855bb70cae9c6d041b9fc1d64e181d8832653684e5c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "136d0e91415b806110e2249f838cd894a37645808f4c5993c863fd6e729c5d92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192144, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s3x3x3x3x1x1_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm100f", 384, "797ac6a5716c3c046294cbc790c2fa731f63b06b85d1641bb341d74b7b2bf064", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "4d98b804292b30e5cf67bf7ea31a06057afd4a4bc5661df8898ec76ffb93c646", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "a8dffec274f1b18a32797149c1cb3447759cedbd19b47fbf4837f03b1e96ed32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "f74e606c90fb924aa3ae5ec36e00951f6b9290fc3c85f51bfb6c4a5435857cf3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "4dd5f706975fed744535fb5b5c195e97e2374c5490fe4b0f65c4d844664159e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "d81d590118361c60733fb85407eddc630e947dfede6f4301d0a5d7190816f4b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "b6271971f5f2d38903291b7efe03509b47129d84a2ddf63deeb80f85c8205918", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "d7c684baf282cca856623a51aab824e4b14ceafcca3507eb02f3da74dcf3fd61", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "0972f3ea67dfde9ff24d38a37b27d7d78e745e2ac399a865c1a9f4271f656a20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "18f25e26ba0e64d56e901b19f7189154219f7b50ca117f9a125b0483ebc6bfc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "8201d36d24e1005af26a88ee41535982119d68e126bb20429a3ab4184b3ee507", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "c7b0a1db975712071d1324a75abf830b1f8bb2c2e6c006c9673fded83cff2c7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "9bb88c2210aa15631f1c8099aa436ee3457b5a6a2910ce02708587296bf4f1fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "b35fa773a5f32393dc4282760c9442eab566050b66b050e79b403d8f5f0e7603", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "804895b02276b89ea79c936d0a5842296e5cdf3f43f92794f446c3e41b80987f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "be65cff94ae09c65e21d85ae8edf41258ec7ba05f962299c29a6ae5baaffd825", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "9234c0b233fe12f7aba70373a86e5ad837032ab8f8fdb7213e79a416234d7c79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "0379f83576becda4ecba3b62201d1d994202444bca4f0cfad9652203475a402b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "5e5f0a061fda955bbb3b4f99d349feb89e10c30d294b8bba20509610668dc27c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "1e525c5c4225939f889e1522158b1e88e132ced8f24c88c9dda3f7a6d6d3904f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "da46bf86b6dbcd36ee01a61d64106c7673339d3cc03d4262b0a8d0ea27312bb9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "a066ca992374b1bc12a93bee1c18f49bd6204fce89bc32cf0aa6abac3c5caf39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "0e5d89f14143ba55455e333c1c335e93462f1f316576212090d614ea5fd7887d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "e90175a9e063f3374d1e4b35b7e594b1262f93ecc6f2445f4f5b91ddb010d85a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "3899138684df586375da8756e315dad7fd3a40f21f78268a352c4935b69dc579", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "2307513552707b0bbc98cc7d424b4fda1bb39c2623657b44eb09800de2756415", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "1cf4b13acfa1200a975c6747e9f9c5b5cae94457b13a07fd1e5fd990321f5d37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "ccd0d9164a190b27e00f5ce9ecdd69c7a78a2adc1efdc969888f8de8049a7971", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "ce6b84d760b73f671c6412645a254ade3c4d0e7ef2918077b7c2b3f990d15376", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "5e65c48eaabd421bd3f7d6fde03abb259dc744eb666627a2e8b5ec2b40d38b5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "2cf6f90d2f1103f339ebda7a5a9228727da2f0b4620c67f4981bfc5074fe70e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "88674ebe69195d98b20d9258dc1e2a3bcc360c331f6478467453324eca1dc445", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "5919437519ad3d33d70496113c2dd3b3655be73782ceb95d04960ef3772a3b87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "ff29cdefaf8a041bf837519df49b598345171e394f9bab799c512b7add13d52b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "e65fde1bdb34a2b2f155fb71d38471a5ce9c8118af0741dcd74798e4007a88f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "ef686b09fde777ff19efbed2baf9daae1ad316c248d2ea48581ad31218d0187a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "e8a18b248508b261a6fb35657df1502429b36dc1b76a0972c042ec08462e9b75", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "af4f09c2f50a1a29e15d4d4162a95f9440f7b7cd71a7c82b2c9179e219676857", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "f9a343992e76da967ded33a91b957ebfd855870dc24bbcec7eec26667feca2d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "be54d7d0875c88bb0576c52ed19ec8b5ea88e694e7cf207bb13242a7d507f24f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "71c4db4dc38918a51ccd681450da7b82e242745a27fce4cbfb749b68bb4c721c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "3934dbae96f61349aa879e242da39c6e9c7d41f372fd76df752e8f7aceeb3671", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "612c31e287994d58e695badbfcd500bbbd229adceaadbad26287ddcc8af47fd1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "2ceb6da0ddc190c495516c5b25c4d898987da38fb62d2bce27dd869e6fee660b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "8cedee730fdcb479690f88f68f9dbb62d3e0861ede45581d131dd1616515d0fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "f8e2ec40c1166d0c046201b79819c4f27b52f533b0149e9cdb8fa0f4e5119c33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "f6f858d3fc7b890cdf86707b837c967810d048d15dd0d9966a5ede5208dda6f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "c3ce2e12be30d595ad50199abbf60aa9df7b24258f9d77b811ee99d453848832", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "edab3da52ab9a5b2f9cd37b43709af9eadfb4ab6bac9e2da015e4c44754f1579", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "c061548ecc6a9e6abb0dddc6c76157c5ef9dae125db27b7bfc8d464f88cd4069", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "badb79e4ce59bbc11bede7ce3dd8708825c76df2f187e7d1d649bc15ff55448b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "c0b4f7fd6e92f7319ef48318f47dc0683e83cd30c7532869333540801b4d60f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "866e345477c510c4dacd4099f4ddf82257c85495563084cb362aa9b004a60e9d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "12dbfc1ec29c650d2e5a9a38ec47d4dddb5030ae9cc1ebea033028e047201663", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "fb5b99e0ee9751438f4a17201443dee3a53212851e79799a3f22d55e88de3c9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "9b747e9fd428d77c3e4fcdf9382192efc301f7305bbae47e43c41ae432709f33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "19f5d17fd765becc7e527ff6d6414f3ab569f31759b15005be28252bc5c28ef7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "ea9e9ea5cbdfaa6cb777cee164c25d318bd6abfdbdf4df1aaa4c0356f69b3b30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "e28175203157b09b791a0fdbdfe0b7a42ba755411c6affc1001f66b5f1b82d54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "755f64c2b50a8076f3cc2f603e7783e0b632054d8b0c0eec98a945b4371b1c40", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "be39a52f1507f9704e488300fd41bcffe064e23cacf58f2e49400f419738b076", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "e24316a5941e1b9425e897e2bd4110d77148490d687748338318fa1d85c261a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "e24b871231bc7afb119cd29fa3e0968e6da6b1996fc5feeb7f070ca4abfa0e93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "b259a32157fd381a5be686a416719f4dfadc57c144a2d124ffc7ec489cb621b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134576, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "d0a5ab5577773a069078a513a11af33c4a4b64bdbcc1678061d1be4aafe4e1a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134576, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "a8133c6f572bab1ee54b81770def99082eebc1e1ce4dd14d8231d46417a97f51", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "24a2a88a6e95aa9e1eb47ecc6d7b6c6e9b24af1d6c9778ed6a14f6010843f0e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "25440985689ee588ea2d73a1037370ef002340794d9cddd2289dcb7c623390d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "a817202d85d7668b65fe68b40ad98a2645b66360a0c67889a4b24fd9e4b3d13f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134960, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "a7c53a6c2d88ad55594c8b04ba698e75424f549acdcc3aea13b82c36b4287cfe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "9dc039f1dc0efc7f30baf19ad5bf0fc682716a1a73209e8832d543fa05871bba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "e51b362b7301497898fb8f377083332eaf61c019147ebb10372955781b894ef5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "5dab3bb4652d18c402646907c417e80334af34016beb863c6fef188b572c531e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134960, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "b5d5f8e3366430bc65dec7c2c35d5a5461a9cc3aa67188ef356c2c65a0b80b2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "e3b3e5f8b4a9481c28e8c1822d7859f721e346e999018ba60f8b42dba1018d31", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "2e09d43334278b908cd6d7227a96f5b0ccafc9890218e265b4ab030a9f03f537", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 108720, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "d85311488daf64edd9c34a546f56e84a0f7d14338bce42300c193147160751af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 108720, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "ae15dc58168644a0be5cb67b9b56afff2d27283c3582cbc10633874156955b64", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 108848, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "830f272f71bfd053c76efe076c71116fae2d49d8a17ea1891722236fa3ea1349", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 108848, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "55f3d5a51153ca489679ed78f32a5c38655c990c429a5f1fb0cc19949e28e28d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129200, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "b34fa40abb5666235a21683848ed42f4b61e514877f94137c760b6aff158a59f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129200, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "cbda7f5d40485e4d48c7792c50a2d3008dd59208314c666f562fe276cdeb1a11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129328, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "e0f53e2eade6fbefb3950bc6c041e23298988b6a1025167da9cbce5bd91997c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129328, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "105972f8d292b5436054faff4d88c1967c8797599732411b5c11be37d7e459db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134576, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "9aa791bdd9a9db1b0a1e042b116a9ec81c58b9966f0b1670c6b2f08d92749864", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134576, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "0f8ee70d21a0b26f5692261c475c47a0a2ffaee25e407c29c0b047ad3ab16ac0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 171824, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "7bf5ecfc495ee6b66256e04cc032e604fa781f765a0d212e43a7883047a0fa3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 171824, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "298bb13883f563debf1ef63b007f4c57cf393581e1fcba42ef308122583bb573", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 108720, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "5589796143f3dd78c4b5364a029002f7926c8f9b5aa03faf1095757449b8e6f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 108720, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "de308216da56bb3e7da1a47b70774b603913df087ae2e36269d3e6493beecefa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134960, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "16e08c6b9f5b439146ca49eb6fb916ace8ea7298426d5f53ab48c3025bd5bcde", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134960, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "f3e448391f115085a199b26e903730259b81b0a533733d51bc611f1ce94b48d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 171952, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "6719bd52247aab674c21d45a24c4b457eea92946ce1c0273ef4725f7b72394f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 171952, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "dc05b5ce19f355f2a836dac0d7c9af34c03544aff17f573b1372606d0b049623", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 108848, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "260fd960fabef8c07e26e3de3d656d56aeda0675ff309aadd5ffe85979db0ba0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 108848, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "eb65ed3f4a3b54916bafe8b8c9aec4b7076b58371eff5c6b535796cc10a99f86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222496, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s4_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "08420bf62e8ba430d005b5d1d0307a0ac82cffcdecd4be88a40437f5a1738552", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129200, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "2f71783401e2f3e4ce561ff29f616152154c4b4f81afe7264445555c42ef69af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129200, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "b15ec8d70bb2cd9aac94813f590164c79f507df802cdecb063fb91f9631ad921", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222496, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s4_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "a3ac21d90bf0a341ce1b7cf0e27e1629e3c84127999cc45ee8e5d945398efc73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129328, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "b3f63dba9e1a98b164eee472400f670c68397d6e7bf3a9d6439f646418750901", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129328, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm100f", 512, "33e7ef0c5e58357f40340acba5ed73bc72e52ac9d0356c985453acd85daffda4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 171824, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "ea72788a01f65c8e82eee08c16a091e4e75f7b60df43fbeb3b1c6d2cb5ca7878", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 171824, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "ca902494a89fbc28c982e80a7388c6803990ac9c918f299e9bc4f9583d0e4ad5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222880, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s4_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "6f51aaeeacc9f9f2dbdae05b986b2273582e4e049230619c65eb3807fc4bcb7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222880, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s4_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "52c95f896deeae32f3edbc36700af8b115aafc8121773e7156b42a0961d17414", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 171952, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "948248e321a9b6f330e90df5e7c3b8f231b5aacd3ef138479851f2d6196c8467", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 171952, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "3d8fe0c98498c6a83910bce32891946e1d2b618d5564487063031a2937404dc8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222496, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s4_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "8bd9b4dafb5dc02f5fb0b8d3de7b840ae2478c5e289e2c22f23073fc09e509d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222496, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s4_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "fd3be73d704c903646c0d582cb95b51f90a2f1a70075fae185d8a0f0f033e0c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "707b263d02bf5f2d19dcfb991d1ba2062ae3e4547815f9744f6ca99eb2e35f14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "8ceff82512beebcad8fc6054d0d8997f35e8daa7b6fd4b0e31d8cecd4cdcc039", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "19f3839c5686076c69df8b6bdf2ed621c2547ccbea06b3b27fbc31fbe7a3f4a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "a8ec2a6b79f5c3ef8f32321d20f489c42ba6bf8a8ba07c1f397d930e1d361697", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "0078183ed4e8b815add4fa4ea3226b8ec5214d515e6029b2f5fecb4f67555047", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "83f6924dc42d2d8a08a248ec553042653295a073a67cab5cde00db39597bae34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "06988c99c171573bea3a49fd7968c1630999b4cb3437969ac6c3ec0c1c2daf98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "518f714f22a1bb5b14f8f3fe507bf80dcd271024c4fb6cb1857ebca1badd5437", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "061dc23915b0022b6be28a6ddc3cf0ba5ec708715df7ea10811d84319ebfd766", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "6daa678a5de61b820d4d6558ba1c6b230be44a479df8a18057884fa02b440dcb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "fdb41f1e07b9759b467530187315215750f491d99d1c48fa497b6dd700b6e48d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "85bb9bc74b91b3ac59e6eefd0dd75b2d5b852660aa630005cda10860925fff9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "4d2680569e073e9f741ae2ad8c1d5781ccb4f82e0b7a2e9eb2abb34f2da88cb8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "bfa345d8ed17a886fe05fb7aa7bf1445f9aa6eb957099dbfca478c704ad0e308", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "31095cdded6625acefe9807742b4390d237cb3cf95adb235ab29c269b8c5cbcf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "adbf00b2455ebabbcd2e5130d17032b3afcd533b1a6d90ab9c5ca4a3750e2dcf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "3bc806aedf43aa562c56b53226678fef2012cd50aad1bbbb4fdf490aedecff4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "3b99660fcc24e23b515663e11a237d5ee47f9d97cd04441755aa0f0aa7140325", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "4e01c3bb54a43bf4ee7f1b7cf492e3e2ebbc01f8b1e9e3b5f7b0271bbe73d7e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "eea3467831c54acdb8d3dc4cfe7da2be2e65e18febe00d5de15ff6686a6e107f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222880, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s4_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "f28353a92cec37606bccad920c54257e3b8f310d1d66ba5fa643790901930fd8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222880, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s4_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm100f", 512, "7300d714d57af69105fce92c1899ef397f26ca72ea4f34c9a3f9ad656b64f44c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "2be2caf8c7db10b9fa13784731cceb6b0aac31baa2efe801e3cdafb1fdcd0037", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "133daeeb2e26078b5505f0c69fb26be76d04041f33d17bcdbf77d060257cce8a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "34b923f1fd33bf99546cc6db496cdbcb25a4f8d3b0d8cd0da0abfaf30fd0b984", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "e4c9442ae54058e2eb7235b0f5cfe9096f094e829140e4dd729a431fc356b0c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "a5320d797959adf746a092e80bfd0bdd981a29b1ccfd7791779da14f839841d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "ef12bf9bd83c9286418319a5318da61acaf7b04dd746822dfff1aab83745b0de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "01888cccf21dc89217bc88fc876f00431eeb7c352dbb887dbd46e07678d41a54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "f21f4524e22f1dbc3c87e154dc976b7f045d473338790849cfbd492ab888845f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "9c3c10faf68e05e3b14e14bdf94f8784b416c7015a714d38c826d93c3250d02a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "fb7cd0b5206ad96585bd6d3ae3a2c4272996786eeb78149079f4ba20094ed7f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "6845dac7a0ff2dbabdf7d2176861ffd68e4902e0ed30394bf57456505d124ae8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "9b03e66838a809160e26d6c13d6e2cff677699542914ce0ea76cf7ac59356f1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "f6e081803791da37e45c35e29168d6b57f8123bd8f4271948d5153bb102ed6ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "515e9ad4dd93f86ee2d279ea984c73ac0c1fc2c774498f00f0c90866c824219a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "7a067bec961d5dbc0c580bcdaf1b6a31706def05b6e2e45036a16f97bc217cec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "97f22c48bb797479965a7f51264c4fa316619bbd608593a76d7cf976235ac11d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "ade4033f9aef65705f6b51f657a675cd400b7524fd93a32e5caf0bd257c818b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "d92a8ec52f16dc107804a807e35b2c5c0f05b45690676e7ee9b048ca83deb008", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "5ede861fb824c311c882c5bddf40c41d1bd0f4b5b0d0e38257c0eebceca268dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "8077703e25a32edb54050dd2a4867284234edd1dc62d16c43ba1ae6dbfbde4ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "4a99b369330171af3d5269be1f9aa62787f2824343547db518a368a7107b95c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "d0ebb126c8d9d7b9e54427fbc0501b5022c761b23aa9d7ec07c34dbc24a3e1d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "da10b90226fe706689a6be4bd09589f5def1b996f45a7de2a68f7c7f01f52d1c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "f0c303698d60a4b3a942e350f7335a19d38bb8c0946cef51b91d800b8cdd4337", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "5f1f116e06452d8b5112f88402d37b7db80d426f7020c16a5265d9742d249eab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "837e2c0b6b4213727fd71156ec974fac8d6cbbeb200cb2208be3feaca0996ded", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "e4e0eaca4afc5a15eda24e3656cbc00483321875e5c0ae07ee8f8d5c3a048864", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "6d8ac109142c2eed4034075feadf14462315aa18cffb25090708975e85b3ebd8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "546b8ef3133e44e1db721ad318063d7bc7dca79e1fef9c919d211720b31e90de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "b441ecc7bf1fe7fd3a2243d193b35384bd68c8e8523e8cb798bf2eb36e67468a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "5c787a4005f1ebae3617e97437ff86ebde5389ecdb44af5af1e9a1297e918f9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "458882931fa288e50262a5742ac46ee269a7854937a5f50f74e835cef626b762", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "bf16ff08b52fabdd153370ec7f312ce6fdc663bb0ef0433ef89f8c5672d50a11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "adade98a3e611723107bde11e814c48c0c0c4f69472d5c685d7cc5a95f314035", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "d74461664d85c976315f45a591076d94366cf3245e33883d8c7161ce8a40807a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "15a448472a644d4f6cf5de904bd33a026c6881264a03069114b799e95dde9cd4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "30456566420139485e42b6c1c04489d252d005c42888ffbfd6cde6d8833a3f0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "de903f9426c996c191568aea81391a0ed15cd808b020f962321c15b372aa8c3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "5f38fa3eff8b45a8122eab24d20b2bc2f7546fdcff46a1757a3afcf3ef6c9587", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "85edae7e9c5b1c6280e2265ded6568a248a61ed3d1c7d172b25388996fbb9e81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "cce2a20a4416a7f7a0391a45df76261cd7b7fc992611e7318e2d0131ac0ab530", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "c47da44f2e5b519903245435285fda5725f96f3e5f68f62ce670fdb0c6013693", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "10b204b3357e68af8987a6e104fe4bd495049835d825ae341f0a0d6c1c68b5ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "4ba1ca543a592a00827ae7e1e8d6b0dd93232e3c5d6b118484c68203d51b46e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "106c29c01104357b404a5fae2df3ae858d7aa6556694d80440c17ae5392cbf50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "b81087fb6c1b8000dc605c5ed74c42cfe8cc51a97a402dcb101116f51b906d65", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "c6204032b4777b2ebac660deee9534824197cefe0c507e76a2850319ebb77579", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "b04fa84ced5cf9b9e3e253dd2b1beb5d2fd3d45c667cf9000aaf550b2f72fd75", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "87e7d9b7e3c4194f0f8a919595565ead7b2ab9087d968d535fed5d1a4f0f1d10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "df3d423cac18c0cfcd6f25359f7f286fd19e8968c29115ae831433d041e2cdb0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "62e160d054edb6d2fd961ca42d555766b3441ae1435a0cb98ff20223b2505d1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "850f8b7695804cd7848400eeb0edd180716c4ba815dd9755d7449cf7a12e9d13", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "13f901a17436f0758ac23d26d9394240c2f94e210321d6696d5949ca5958c0c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "2d45eb06982b2851d5f8fe6e309f59953f844638b6297e2d7ab4bb664d1c3922", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134576, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "6362025e40324f03381d59b5fd14dfd06a2789a5c296a82ca657f88ffb08f604", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134576, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "b0aa62e22b1f0f0c9d4531a1cc7752c32d0f8bec72654d26a4d28b2b127f308e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134960, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "2732a1311b8937815f49c870a56e00ae1f91a33779dff2dc926b566bb106af95", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134960, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "b4967cf834adf3bd47436b0531edce50fdd01de573777961d7a0e3ae5c1593c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "72f5be094d77a0f7fd011584429f520fec8afd339c8ae2daf99a86d0e50bc63c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "3a0d884d56e1d8b0dd3628fcd324e0181897ce5bf6790510a36ae3f66e3b5ec4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "a11509a70411babb4bb45333509fec6c3c2034731e0b326c9656535040f6a23d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 110768, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "945f5d9b6c2af10ed6334cce35e1a35a4d24bbd8d228e66a16cc11a2129d1592", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "26dc807fd5f7722b03898e856b3d730d9b011cae41c26d4d403134260ee0ceae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 110768, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "502a479b16d9a16672aecb2f2311e6f79c1dd96bfbf6a6d2d4ed5de7ac5bd21d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "5b4f6c7fc53ff5c0ae3ecd88a88c2a680aadd4aabff6e3dc9d8f1e47bc15c9a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "46d0a5c4930cac506de099f815845a4f455de1b9676b7428479d1d656bad80ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 110896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "fe727315a2bff04b58b22491a3c466b1fbcb6bcd1a915dde65f9daf290709670", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133296, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "7300a1e9119b24abfea4d18d150a44564b995b75946f72fbb8a697aae315c57a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133296, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "796d8c7dd857c206efcd327c0f5f0a05be04f950592d009e953eb6d5ef72642a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 110896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "c0af94636078d3651f1dbefe74f088b8ab7dd8d0ba2306361016d98da4be0350", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133424, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "30c3f4c34825275f7806cf4f7f036035623610abb9b205755ca78ea6de0b3c1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133424, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "68c0bffe18ca7fe42f0968e1c37263266ae08c225d00e2792560c0ae01331335", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "fd6676a19c87e95da6fa28c81e0d2ce293757ed1337d1fc29ada6067fbdb6fc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "d8772a59aadd4d933d8c70bbc8809ecda7da2d411997197928b0ab5b8887e1c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180144, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "de32664874bd0d0c2fd0cd240dbd588dba58eaad4a8329cd7727a2389bc99a4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180144, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "4d14e8bfb4575d31352064e0adeb61fc5a6c969a011a4ed116c6eaa026aed5c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134576, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "e74c504ebb9196b41b2a72c7182b490f31aa69f471e96f6828c3a8584a79cc0e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134576, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "390bb611c3ae3fcdf48e1c83c9db194ab3ead51e679c332bc93679ef6708688e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s3_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "687bf9eb92a7b1f7e841763485f96e2dd3119dd57f0a2333a8bf829691c66172", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 110768, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "dfa6450ce1f964970b17632b9d3e94654315c013f38a977cd530be8696cf12a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s3_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "9a9877104c86dd3b5502e8ca957d5c57a2f12d3fd6599597d15735a3eaf67a89", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134960, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "6420d669b8d4bb31ab85a8b9cdc5c763d3634a1f1dbb91f478735973f95fa875", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 110768, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "bd464b502a5a964bbc919651bf92839ca70bfb47121ea1cf96b78ff81af26bd1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 134960, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "b0dbf246842ed6ff62b4e9b24791c43b7e2dbf4d97deded0eddb795c12582f36", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 110896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "c8c6c74c07e4d4746eae49cf9c1711c162b7ab7b59094397993488c46068c913", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 110896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "bd9540f68b9a5022778c93d8326bcfc94ea67e4779614b86f56b5e331da97ecc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133296, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "7a490d9c7013823ff2bb997b0b4f0ee36323306c98481e9228d9437a48fe16ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133296, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "0fd5b83315a70a2746e834f992c56bcd7cedfc9548b3d4e748ff990dda252420", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133424, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "085fb1c8fb62a57b8c7ac294d4884af4dd4ca855ed3f68ac34e9c1c8cb7607be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133424, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "fff0b0d67b14f7e1762ecf549c074ef7c54bb86111acac83b54bc0008ee4310e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s3_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "11a2b91e815a9e530ee0a29a55a346ae63c3cc4298c9b8700df13b33f484ef83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "c493647574656cf06bcaf57c9d79e86601b5c1a4b5dbee72a4dcd7e079e91dcc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "f99d24e8cecd3a51814b7fb4e76bc76ad38a5abee3c6b7887cd92c0638eeec08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s3_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "69214c769c70df6e18cbdca188e2de7c91229d165035a661f98f54387113e9c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "495d19d3e48eef5100fa762f3c41c09628e81fc53c1ff8272e8f5f41571ccf52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "62386e633b38af43bcb234491cd5a9c0190939ec8a644957a3d416af4399b2b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180144, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "93bb1ba36cbc7ba1c6390dbd59014f0081d77b5c4f37bc31f384558b4b271680", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180144, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "fb0562175e67e0eeabe1476c56ea400f4d79455cbd74b18fe9104b642dec7154", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "aa702341d1b374a409859cde5591a991dd2bec5c6c28ef317702bc6d16a87787", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s3_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "5720188caca98716f7ef051b76d7afded87eef313100720f5e67396d8e2b471c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "d3728318eabd88ad82f6e6beb5780cc0b3992031c2733cbfb3e4c918cf323509", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s3_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 384, "489d90fdde703dc6cacc1af9831fb67564fbf9fcd7dce8f3d2eddd19a7d0adc7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s3_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "fe28814bcb79d4afccde29d9a64841c68e0804aa8ca65b9fea072d59d9ad23a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s3_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm100f", 512, "5a324471e4f9bd4a18b340240872066f6dac46c5b2a61fda6f3629f856f7f91a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "6aaa062cc41c4dfc79a6e71a8fd4d4283c98379668dad2e44d12bcde26a8d5ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "b9641aa214f079c89293dc0d4ed04892e885a2c250b7521d90aacfc5fa082469", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "63a1dd60bd90dd151ff04fa50aae94555fae94f6ef521c3f9736edc1872a7781", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "6b5e0a355d10e5d0e062e641d1c127e1980e8c20036e3113e2f8ca26852f3fd2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "5129aa83276926579e376d7d683472243ee988e6d34cc23bda870a2dc29214b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "eee4b032bd59adb40098f937d3a7aea087adb09f0396446c096f8b3b764a4236", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "9895083572bd59e0c49802db997d7786524e3f8090cfc1d32ba22ebe26edf87f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "ed39a13826afa0bad27a43cd59ed7ac2db665efe3ec3fc5ad89cb9df7de6c836", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "1e4f6b3601d2ce4a940d06341a7c0e61fa82e9e9e74b603b6629723a973963ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "deb858facb7bee6664b38811f79cefdcaa3e2923ddd3d5f4e78d85893c5f2e7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "9b9e5bdd629b924b6a9e9097f58f25466c7a99321cc75e52c031eec0b06a67cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "b2d1890f346db2eb4b8ffd108115bc5ee590d14a28b4788fd9647fa454bb198b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "a7892a01764650645dbc3e2085e2365667acf790cf34e9612a8e6e8d6a7b9d30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "e5dd5a3af8482a585ab939aec56b3cfb54f20be9be97a64d9925efedd8f7f758", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "650b907228e6c80b126dfde66fc46db97c8a590b3ee50379ed03cf2faff92434", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "1fb0c88f7f9f183358d76a9986c0f585b3b10f78a757590a79cf46a5dcce8371", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "15049bf3e4ef0892cedd1bfcfff2343dee92aa88639febfdc17617fc3bcf2ebf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 256, "4822cdee6efe4733da133f1a6e861cb40f7a380a0a6f6472f7ebfc9805f351f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "65406159a12fd41703bcede2ccce5193cfe0536f04cae6f562f7cfabad4caf50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "beed3f76e28253d16efd42d28f0b52bbd7ddda328119a318cc6390b03791befd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "83243cb16d72076abf68443a43c8da9bbfec416fd4c8dfa601b8abc5115eb39a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "cd9d105159a2161bac607886ad626fb6dcf67ba9662b40112a85dd6fc845d92f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "f78a49356bcfc144c28a772c4d072cdf566be2afbf47eda75e636e3d8d324409", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "b1d9dade02d821c4c211546f52d5b2a3fcfcd0bfc404e00582b7735d9d20fe21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "a38ca62867cd06d48a2201fa149c82cc70c974a89c9039e2017293bb17f2f7f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "914426b30e8db2de1bac1cedb2568741c8fa134314e3d36d73e59620a1a12709", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 256, "76d4e0e821e911b83d3ac4b434a5ef596778644d4fc1545e5deaef5f7ae6b276", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "561bbed9615442a6f85e470a8cccfd84698fe264d84ee595b6e4410f38fde28f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "3c60f52c0a36ebd6ede936f70152a65a88289a80aaad57113962ba091b6db69d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "0ed58d19cc89e9600b7c64ba73f03ec49d6263abae78e87f05f147ac8a39062b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 256, "29de857ebb15f2d3206a915dc1d840f678e7958e6336760bcb979a5f009d5e54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "065e2015c8cd6c5a4ad89aeeaed7ab736f8dec135732c74b286dac5fce392a90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "e0766a6f902e3cf755517c2a7953be3ab1cca867c2101062e6b9c39df5af0f9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "b35510f877d37dbdfa40ea90db6c66adc02917eaf09670bbf56321005685785c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "6f5bec21eabe633032693373b4067ccdc000ce6d83a2181c7556ba0b66e2e5fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "84104bddac1e4e36c43727212b47317e9fa5486a3e4adb3a86608e4865fbf1c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "e35d932d747d6b18599c6da614708998399bca62a9517c8994647c9d8518a717", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "97c6ae646253d2861874caadc9f72a25a94757e6ccda853e84016fe505086a26", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 256, "ba0c17f9dc5ba4bdbfb8e24199734ac4af80f9ae218f320d8d0afd43ede456f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "d5c3f9905a25c749c889e924666df838c59a5bdbf12cae45a7c801cdd5414873", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "a4248c9c1516624719714f26b779e0fad0912d51624980c1032c0febcee4b4b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "0349a5a1828b6c01a7e88e7a92d26b6a19f0cbc90e43c3b22a9227f72f987f0f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "f70b5d587d161da677ef802daabe8033fd03ab76f8389744feee9bff3f1e1f6d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "3d9ef32025d9c42f65328ddd37b429b7c600427302a4a9b82c9c26a2669f5067", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "14efdcc096f851a686782f0383a502fd8cd6546430aad8c346342d58b73a831a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "8b9750c5e75e6ec5f7dfeda7529a15b7e9c6abcf4e996950fad0872097e78cf6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "d41667379ccbce46de1e56a99dfdb013b4239c609361a1a329f4a6000f690153", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 640, "f4b19e3b4c9b81de65a2b7ed5aac3f31f30cc8b4c289a336b7704f82314b234c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "1dc84508bd513e738cbdf4426dc4486cc6191e82d313f0c0e056c9881b3a23d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "4496fb763be4f72b8ac96572e0e16a37b48b990596f3106f6e5f9c20b5cd8449", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "ed0fff5fe0d36f39fa6ed0e50edfffa938be9f309a32aa670cf951ca1af14b30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 640, "3fd2e11add357c2ed67c77897192596c66bb3ffe7604a0957c870f1ebd2efea8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "b2eebeb0968fdf631e4dd862c5aa8aca453f2965adb28bcf2c10a3f69b29b93f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "9fc25cf54095736317849199e9845631cde489a8a02e24a6edf400ae4f0b5f50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "87fd6c18551166c67f0e3f84f2af47665b959762e2bb101821e210be4a2aaccf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 640, "5e6ce455b151080648452724f9588e57862b71da51e60a84c8df943fa84719cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "56a543ece1322c363f0b0cf0ddc6aa1cc77c8b44d63bf5178cc62fc64916a66d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "d4eb254dfa920659f8be27b1a97adf08bf57d65efff6412405d22bd4f66293c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "2387c65bc629b7a4c3aed1b7c195db0b4efa27573550bcc53ee62a2f3549131f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 640, "5ece11e9e42da71d0654190a41a194ead157f819c0c0afedde68a673c4a2cc25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "a4d2955fedf7440420fd6579f10cff1d3b6f6f9dee27cf7a96b03670f9305f61", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "8c3248597283e53c1b11052a43ab7f8a7949fdc443faffdea765958866e01840", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "7528a452adafcfba326aca101f3ec05dda1c37390ba23c2eb04d9b099db2024d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "0328c8de06f628cbde03d354778f4b38897bd45b8ac4c00432a33e939f28109d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 640, "04ff8e762a1bea96570a2f489cddd3a6a782988ca73a763b29490a562f4114b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "d061e96ce848718503100a4bb4a59066c22145add9d4ea15512aef63a5180f7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 640, "25b98270b74340f7b7b9f1265543fb78da65f006390175cc25061ea1e8315020", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "e5fee302272d9f99f6dd525b65e2f2699f764a4bce138549f63d716cc6ad1e55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "dae11fcd01a87efb4d71183600cdb5c4ee726daea8a5fcd3739be8eab340031a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "017deba043022df39230f14fe778a8ca3efc82c24c7ebcb1a6d0682c56ceb2cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "0ad464a08182309f418e861a7e5b2a4ee71e58c396008f96c48afa3a244c6bf9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "865fb3b696d388d504e33971e6e8b9f056bc0d32f63e8c2a2aa873e750b1d333", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 640, "11e21992d1ec55c74bc84a61dbaf11d7c57b42a1fbefbcdf0fcf88277f914a13", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "767bede520b56c8dacd94d1938830c7c3d08509ec1ff8f5ea0020a3473aee4de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "4bac9ec16e45d88769db38a7434addde0c15d175526b98da3d081cf5e61a0314", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 640, "4ab91264d3e0d8aebebc03b5400e5d1fcc18469d463206b9170210d059fa0efa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "962b955326013e74eb5f55d46e5c10c325532dd71b6f36754dafcfc72616d51d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "07dda5d83179780cfa94f82c468f6b056e0a999fa6baea3b21539e443c5ecd06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "b7ed5c98c74757eccd42bff090e157ca8dc8aec6155f24f842c86545adb1e938", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "f78ce7e8f2fcc40f6a7f3301a4e9922ab6b62c3894196fcdb13c69940744798e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "651c9c1c4ade058526e65c1143a2f4822e7abcbca876130e8ce6a85cc14c48f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "3a848a7a553a54c9e984d56c27e11bdb0a0d4b519ee9ba74e638c7ac23ed3030", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "8e7b8b667b68630acbbff0ae07a9cc71b6a1048ff3c7c12d4957028f669738bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "a96bc79a4927eb46171f1cfb8270044c1176afd8708e8eff9cf5dddf3bd86e09", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "32d7b359885a34be59b05ccee3b2503b82eb71c20ef35511fc52e2bf3b7c9c01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "1df184114d1ef47b16dd67f1a7500be049b79acabb9c2ffc03cc55d86a3b2fc3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "b159227ecd740d9dd61376875779e35db3e1c32a7c528e7f13512ec7b3504b04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "9550667da0a867f073a46fb51fa2e90c5190b43defff365de84c544730b10f2c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "38ee599962705a20b005e0867944ab01355247648706c99ab4dfa49c627ed8fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "474fe2ef50cda53b769150fa6f1107a846aa5f83d7590a64150cd080569e1311", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "7997d854b543d3fb573214159674d579133f2220b1c27899b11085dd1bc36a13", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "393851966967e66a5cdfb253fe9e4e4f77f7ea7006fa60451df5089274ca9aca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "6a8928e87ce2a4118090340c0bb566c5bbdaec35f2a52a576a758734d3fd4f14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "27e4b7f86ce36dbfc69518956101dc103340e3d058e2742c29e72b8725e8d7ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "33baf5bd23538915ff64a51d4207dad36bb72fa474de8f47e4cc158010da052a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "5f0de907ccb75024eb974f97f20d33e6e881af2eb3f2e28ef8c27c2b9ff5ec63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "9bd0697605d34b8fbaa144d855f2b09d597502bf8a2ef083cf8618d6384b532b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "28c9c6fd236e7d8a2fd750642e9f450b5977b8df828b6875196aaa4a7fe74784", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "7e1e9008c37f54d839e76579560022ab9bb9e6eb5722566bd20410962c647fb4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "1120b61ea5dc5ec9e8b82cd0243b0084dd39bf204194b6b9c2c825c165b9fea7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "cae7cc5376ac0fd607fee4f21fc2c0744f0a7bcbd29834c8ddf6134a31810073", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "811199616513505e4d36d29388e7e0b6cb5beac5a8d5072c704948fb5321e4a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "efd639d47d08bf0562001a749a3b9586a90d01e8ea6a19a125ffce57507c4a59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "ce307cbdd46b5d764839c830295e7b4b99cb6c41f6a0dca9e38c9c81ef9c6b63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220824, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x256x128u2_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm100f", 384, "acf18a1d1f2d2ca320358299dccf02dd92fcf28029444b46b80aaa2eca0fad14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "0bce404e9ab9d369ab594d45d1ae791962571fb25dffea811186f37a3fbbb536", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "56de3d4573b7f11ecaef33d4df312c077204ab079af764eeb2089df2d85e8de0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s8_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "166c6481cb6f95682daa1055b68acf1269e76408a13a473f65df4ff8c6e508e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220824, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x256x128_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm100f", 384, "52149753f017c9326a67a837baec0c7b792b0e8a80aae3f94fb5d879a31c2707", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "437e0d781930e40c0ed99a4f5090350a07c5ff8222482a78c105e9804c6b9d89", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s8_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "4e3783143a4c024ef16331079666289ac7258932a667278f161703ace59a762a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "04ac116bbbf47d9936d16a78fdbf9961223bfec6d7b20af3c64654309ac29b63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220824, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x256x128u2_s6_et128x32_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm100f", 384, "fdff33fa665093ccf3f4d4149dd073c7c15c359ff8a0e0f23da57dc06dee6e8a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "5b2749ede066201e26cc94101eaa202900d0d6822632e63c41644a5dc5d163e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "841356ce0cf72548897c2a4659d67fd78b58b5a8dcd1f722e6f5ee62597ba257", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220824, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x256x128_s6_et128x32_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm100f", 384, "aedef6a8b5f7456e7735f6c6b09b7bc3b442ce831248dc6a31060e277531fb36", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "efd8ea650a559f81ce61981664d4559f5b705d4daf641e597ffac43edebe6eec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "dc1d7b977e4d8353515528cb6cb14a8f32cda3c6e3f381b923916b4270681f5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "b7c7bc33c2b535e3fd050e0be9c4de6048ab199749d12b5376a30bd107ff92c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "1473653a1a7cdaed1c8d92920368029776550d777adef9491a49099a959690a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "85de731124f967cca5a457ea3e101e9d199168e7ce04e5f25141f3f05d2c2a6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "c77f2c4e711bd91ba1c76ae8347ce47928d59b81a292e0f59be37692f245b9da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "d6ff1a251ec064968592acc679d8d11c31fc5bb5f87a0b3436e3dfcfad8d956b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "1343e85e7e235b79aa0c1ab7473a3a8e3e9e2975a8f0c5a35a0e6c18ad48d46b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "470fb20a9a3c5ef937562783b726589a4b6f597fea7d56d04feed3b853d8beaf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "a5c928a5f3bbd5c3fed1782feb91e4010e115fcb400fe0ea49829fa776a97192", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "d7c750bbd2ce8221977fa7aacc00e9dd14b8529e7ce822b00fa903042d3adc18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218384, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "57be854f4b2fa25a71759ddcacfba98737d73c2ff246a2de1fbf7f4244c97c5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218768, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "7a9803e49495c181b797644d8f94d4f0e2ec7761ecdc55f42c11a2c7cc515ee2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "ea644227e814b779b0c011c5ec5231b96b2bd9e1a43429e354bcaa2c3b3888d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "421a3ae539f2a5dcbae2c46abe46b278feded2404e8dc9cce7b3d4e42d3d0b35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "24322e6a34a7ce59b26d522b010e7d7042e3109d4f7e5d5912f131009bfe9fb0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "adb355f930e1e2514ff115aea432934653a8b87e9752b54c236e24fe100251ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "a9b370d71013bfddea74b6deed6725d955cb053ca2453e25e546f4b1952eead5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "c1b05f545608e970607a773e3f3c64fa33b36c033bb0a57e64c4c520d8179d12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "25b5cdc87a2608ece669bc4fd0222a0fd0f6fd4f78bf621d565dd77f00fe8be9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "40412f240b52e8741a1c0cb692c436dfbd795e0da17e90a8c075be3ad6bd54cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "4f635906b7fcdf3e067cdcdda46591206c08adf85bfd9c6cb9f16b3789859246", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218384, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "8b3216accbcfbf8cfcf094bf8acc4fa587b8c1db6c6013a3a6246810c7390d44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 256, "3c32e52c53a91a1c61c8496bf6e2c5601307008e3537cb20c5ba2cb4f79916ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218768, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "84c1a3dd6277679988cdbb549bef6e4262d066bc39faeb598d28d84ea3e597f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "84fcf7fe88104901d9b0fb75900a4ee38011a69f9953909ff225d3b2ff86552f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "f02f340e3701df25c1ccc5013d048b61a7cde54f79236fe4c246734b46d861ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "3d53200af0b08d7afce6da3a70111c4cc2e126b328e79765da1ae8d2e65d0313", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "6c99a31855f532440517a247a3f8bb6c281238cbe22548869c8d8008f203b985", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "7d64c292a61ec4cb9e1397823b28e0df76cb31e31246f8d49bdbba91392e417f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 256, "fd520676a7665a466d8a4358353090e9f3b6cd0b69660c96435c25fb26aec093", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "bfba5c1dfe687539751d15c1cc3098d71005e02f371cc63a23108e463ffa73ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218384, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "834a8fae4003f4c1b6201b494738071ebe53fc45bcd8bebfe9124d3194b819a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "c53d04e55519c9487cb0637fb8c2dd89ad9178c6fcdd32efb56ccbbb20b5de5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 256, "4c89bd34a07ef7119b4da9b3f33270e0eacbe30ca1e5c8c5a9113db98eaec543", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "b1e661af9add16065819c96822a63d2517eb1639d3d02250d7c09efcf9a888f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218768, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "89cfbfbaa6d6d83549adec801474c2a4a698c02c62cf017192471762e55d2e47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "5ceec588a14749e6c556d4eb9a8d37b97af121f875cfd59921039afa04b5f446", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "7e35a08213df56d62bc59d8e64073fa65bc69320f283395a9c52fb239f801daa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "ace45bf4d053811df7b57773da97818efba5104f29d9dbc1c2d9379bae46003e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "e8504e25ca2fb4552e65f63db0e66e1669a1edfae4973c8f0f94672d2e61382f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "a6597cb11e88cd935cb8fb39a7d7e2ddabad639cbeb59174d408d1130235ddd1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 256, "d968e0e37ea8b3cdeb96533d37476de2a7b4ea5536fee0d15c8823d38e371f4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "ebe54fc91d70864f619a7ae5c2148a91d561cdabca942ed6a97033cee96f3b7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "fb5aeea0919e91b1de061b9ab50a1cb7af08d23b335a1a689cd9e789be7e3c60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "45de1f883526515cd812b05ada535108a58d7ba12d6621f8e38435a4b75b1f38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218384, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "2c8805350b2b0aa282e1a82b029c7cc5d2b98ad337886ba5c4db8ed012654fe2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218768, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "720475f87f6c64cb5dfe843b025490d1c0adf9ea51bc753f3706fccef84a29cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "b6b92ea7d0b0788270455ff6b175e97788e408d9bd6e537f325a730c2a1fce25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226968, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "62a7a36b11cbb9242638b17dfa92b73149f23e5febbd8a1bf10d5eb047e45b24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "0e9d2e70022a9fb0a097a45c37205753d456cf0966c52e0a5e972d6a9e12d754", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 640, "bfed03f1c6b2b5fd52a8052429998573754f7cf2f440d6c4e5b3d133e191becc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "4d784f739ca28771bf369e1fd1e01e6ff5ee0b0ef7e6c7b16dba6c8296ccf9dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 219800, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "4ce70dd388a11271550f0bd8090510453c07dffaca649373623a40893c8b1f06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206488, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128u2_s5_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 640, "3a37a06aa8d9ae8e36bc48edb584a627572803f223c51f6ceef9a23f09c20e20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213656, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "545fc87a689b4aeac9054c6427a5b50713cd6cc42307aa21c339c8c7a4c7598f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217752, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "b27a280925412c6670970174a295abfc136eb052721e44c25255d49b8c6590a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226968, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "5ddc7b66d708224df1776d443a4b16251f7960c2233491dbf283c1024251fffe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "e86853a3460532d38ee0297a7dddd6bbb78a520ec4ab637d02cd08d2330726d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 640, "8b6e98ce2f95b426d3957c5d54654c61ea3863cbbdf330be789dd4adfe14254d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 219800, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "e554abca98f9ecb231bd042d657a4fafdac856a465f3ac4fa121de5a25448296", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "337d0c2826ca4d573eb756bda677bb34656daf1bc0a12dda3df52081c096cb02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213656, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "b8f11b743482c7f9659c4c2764843f7b938ef143e08e9f20ab959e2b54b0c8e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206488, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128_s5_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 640, "e535ac5dc616cf03c61777b21ed5cc891caacdb48a1356884977b632aed85d05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "4029b3de709be3bbe478fd236dab974afe738b93a4e69fa993913b3725b87c69", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226968, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "e059d817ce53405c29b0f9c40af790c2ebe75cb997361dfa279dcc7ac0dc4df2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 640, "c5cd64fa6a61aacf3d1357cdcda1f1bfc4306550143b9060aaa4aededaf58a3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217752, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm100f", 512, "fef64a0b89dfc9adc01bde2e5a116113ed76218d3f30c8d3cbde4bf389373476", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213656, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "4c276d68b9bc4512141c3e196c65230bd5c82b259523e54ccdfdadd604450cf9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 219800, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "b406b4fbe3280e12c9aa797cd596de6411db9ef626e9b0e06d9bd79b6c8c983b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "b719e394b85a5d18922de12c10683ceed007c6ef636fce9832ee8958c1d903cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217752, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "81fb75a5910397b447c2a59ee588b7a4b27173a760a3198c31e1c3d0b2e66719", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206488, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128u2_s5_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 640, "4971065b38c52badff00caed3714c32061ebe783c206744307d61c95e89a2b8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "54d96d0acded1d133373bd9c19886e5c38949475634760aadaaf92664708c833", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226968, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "d3a6dd4fd81a817be207447faddbf92e184bbcb8c0d41823aad9b7ecb230bb56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 640, "252cfc571fa7ef0f9db67a9205be844b5f8a278560121a3f645d795a41136642", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 213656, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "b675fe09cf9983ad655cb8e6c48712dbe8c2b506f841f125b5000d01f197ad1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 219800, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "507843f35f247af22ccc415d92d4d522dd2bb454085b60672694b4858aadb324", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217752, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 512, "b4b451581ac2dafce7e86b1da56614c38a01b0a9c7063cf05508d871adfed7ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "d95e04d303de078ce8e097381b0947f6a20563181b8e6cd9a39b2b799430c5fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 206488, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128_s5_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm100f", 640, "356a1506d4f5763d19dd25b9bfb3a84a6e4f54755eade75d806869bc1bfa2086", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "af07422796ece4932445a7cd0b29f0fb0171a60f82e941c995da9055382a312c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "ead5ad1807379be5068553b6c8de5b453f1ff240bd9b06783c632a9e54cdc420", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "7ecbdbff564f9e782fdb6e63ec67f8d37493e2f52c42edbf97a705587c0f150a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "ea559f395267243f1162ed9fc675d0791ff90dcc9cdbfbc4d29d05fb2cbde576", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "a739712ea2c95549f115dfa72fc127b23c6683670f22173c04871f5967debac3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "dbdd026609a8d9b68e3636c9bd0b3f89fb03aff669d504cadc64fe8b362108e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "5fb6e144d78c3c0af279c541915cf4f0c9d417ed881e2c1ff636d8936f2ebe7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "ec7cff46e59c19218dfce7cb6d311cb94f6afb3a120681151e32d518a5dc4bb8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "5a753002bc6f1d191e2d7aa3ba155cdb8ba2bd5d0c7d0e5f7e79efbb5555a3f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "6bc2a53265e8effb2ed2c5b09d80b0f69c256fe6d9825f95462439e6500015ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "1753a77afa2cb8fde64ab6b6a8235fe3a6626a7bb20185fcb59c27415769cb9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "e924a47959c14d09a5c4b29e22811a2a01eb670b0816dda64cad1e88dda08512", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "b0066c6007182ea0ffd2f3470099809c9d2be7dbb8eda8b7267fccfb79e6a107", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "4ec8a2fc7df2994454b14c20d91e843797e9bbb09cdd2f55b2dc2f9df12d0792", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "f4e620d46c5f7a7ddf5dcaf4f3640e9516258dcb7910b9b5e5e7246583fb725e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218384, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "d06d97effca8e465062c66931db52fa507aa2bf5028f2c86297d79fe9d2daefd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "5754723958f2095cb603fa68ba689626d7d10e3a963aaef76e71c2d947872ba2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "cb035803488f6fd1ae5f9e90142e351694518ebb4d180a63f2da4f21a736dbc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "3ba0d8fa11653d1f88cba8520c1176fbd69f96fbaf3b8a5e501b062d8bb8faa4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218768, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "ffe893dd11b20d12b7389708b217a094a97166b9b8323e49fe70deffdb28bcd4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218384, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "be3e70656284a02efc382c860193c92d4da347accae1595064d92c2363715bf0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "180865e5205be4322a6f63eaf9e0269bc3f606e239d9cbc6243ce81e22dc0ee6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218768, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "f15c29a77b7d9632b6a0dc5689200976f2f031decd933a866c3c679e8f83f98b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x64x128u2_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "d2c49bd2cf74d461bfb2e6a1f2361e687420b10a2021c2093281d01d6a5a1592", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "216986a3e9de3b45e30ddd3916729864df9a1accf542c8b6fdf46830e5a36fb0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x64x128_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "33f143c4d97ba2e0bf1032c4e5db9824b719c580660bc64cd0fe34b882b039b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221848, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x256x128u2_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm100f", 384, "131bb1077ce2ac7af1076f76956174fe24f273fb6b9f45ab17bde558ba2601f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216728, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "70a657e35c118219344e4d838d24db2514a9efc7b4c8843354f19211ddd55d16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "09744f11059e523e1e0d136126ce41e113e89f9bc57a9c8b808acb3c51de3cc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221848, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x256x128_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm100f", 384, "1e0b6ad5a0919c74967b345a808e8fea70ffa6788e77e2ebc787f7db7e863f38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x128u2_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "70ebad029a9b0f839fef1f9e05c86ae9548cccb64073fea56fa8af8f7e4ef9ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216728, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "05491fe2cfc4e710eff3aa194e81ba674338a39023a1dd30fa498bed5abd5aaf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "8fd93438116982e1ce0303aa6abfdcaa7d3606664dc7f14846cd69261674a34f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217752, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "a1ed6bf0c5fa057e650ce9a7cb12b8d401046964547fa0559a375c2530983ea7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128u2_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm100f", 384, "5ccfc5b6d03da5f64be9614899c1efdf47bac79f495c9f9e7f6294264fcf4eaa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x128_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "2ea5da1187bbdf37536184932244994f618de374011e4ba9b981a487a828e67c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "ae3018ec238e7196495d32cf8f134ef821ecfec58d28f0e86a7a03433e1824df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221456, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "a137ad5cada67851d4944024571fb5da208a1d51605821d4eba8006f274b9399", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "1b5ac029aa70695563700f9132e438ac844e15d229672919c90bb2a4782a1bb4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm100f", 384, "1358e086eeb7c0d093f23a941f1a8632eba144b088c88ea05d130e4f6f39a200", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217752, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "5501dbb5be7818f6b28070da7bbdcd53ffadc76b2028803642e8b854c01bc7e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "09d5dae1051f3e99950b52126fd8363bcd06971e551b49f5f2b003d25542df19", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "10f5758ba2778d1a7839531fcaa15307f2750d9278f76aa039a0b84d2c0e042f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186000, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "6c299055edef6d631f6e5b9f86365703859fb026151425b3eb6f40c4cdb4d6ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "1febd6a09cbde57b7f3a498791d2baab7cf38f0dacd95634be374fd7e0eee852", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221456, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "92285e32ad6887178ffe04df5bb31539977624249b345b21a45fc7afcb124222", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 219664, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "fb290f4a1c537b030a82e64d0a7e54c01d802db525c75df2fb3ceb486d905588", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "e2452b5ec4d42342f4ba4d5c4f5c7e59bf46fc6046b27f515b76dcd3978b6e28", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214544, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "db2843088a5803f32d38c7faf07dbfb22a7aeb118d4bbe63d5a654e5dec107c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 219792, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "6eae90b51d8d7f799ea7630f4480cb62e61abe9641451f97a6dc15add57d63f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "f394ab16c2db7b21bf6780042178b54a552e8a5de48b8d8a24c7deafdc5135e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "5eea32ba1cb8da02d1c264de7e3025fdc8612a4ec32aeaed2c0dc106e057585c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186000, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "1d052727eac422288771cf39cad12a9638f1703f3ed7428caac0d0e1114a70d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "33a9a52526a4cbfbf69903b43341e4d18eb5cdb2ba712b05a60912fde702aa64", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "24ede6a56e41815d2f327119ca90f1b8293c9b95fbf5d5d866ce18200a84f26d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 219664, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "4b90da440f3bd78069ddb5f43d686a807d54c89ee6e6257a49f41956f57521bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 219792, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "6e7a7f767caf7a5ed12197fd862ae9363948ff74d45eee22b93e2d85ef3c6efe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214544, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "bced0548f1e51ffcf1bb8d65dd330788b7a875947d9613b03c403ed6ee76362b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221456, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "232dde52c8510daed4b76cb557e023b10edc0857375fb870194b887221d587aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "6d7a0ae631e7b2860b2ebff491f75c3f158a3282217f763896a1cbf557a9512d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "5e9ad6bc5192084f887c0535a5fd107084fac7a9cca228bd73d730f5de44cbf8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "e8dee39c2ed2e584d6d1c0a73dc761d25f57779474b341c61b41fa069dd763a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "94465beebe69c4bc7acd6c13bfc44467ac2a61431de81e6d909d7d3115fa5e45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186000, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "64f338de1354af879ca5199c7ab0e810a08337925b007371d714a6235215f4ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "e641623024cbf3651471086159f8f853962052f5f4e50f9f468ac69ff7b8a9ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 219664, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "7e42e1ce2c6845f916d408aa28cab5cd44b99e15072e8a8229c4dfaa97f709f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221456, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "a8b359bda86b34333f477e0c12b83a758bea654f03b6c10811eb3f8e21575a2c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214544, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "a6508e983735b80d914dd68288d9879251e3d56770888f3c54916450f73aa739", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "2d8bca4051c4b4813f9360428382b0781c6b9fae630f8d0737ea8d6613b51500", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 219792, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "155c424c5a1929ca1476bcb3586c76302033447c68def2e699494c4b0c30f16a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "5a8e189bb000bb9e7ca74bb01135f101046bf18a5f9c0d7c1b37efcbae8db2c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "cc06753f108d18ae295006095cb726612b35e72929d7b2061c1829d896cd617d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186000, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "e9864923475f647a0342e95236d7a1ee82e15313f64eaa902a3ee045d1885fbd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "6827ea0f4105fa12f7479c3643e05934700172f9e56a83d446d95fcefa3877cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 219664, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "4f1905f53e9f0d546817557ffd557331aa251ca1cf4249cb4ae2ab0e7f19487c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "4fe052ef593c25e58ebc5cb76506b4e888a9ddf42707e06484f8d237dc0e8be2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 219792, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "b751425bd876a3aa4458d94c60f9c80ecc9ed7d43a0962a62b5800b554607ae1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214544, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "acd7a3bf4a1fdcba6ce5b15727e69d8bab4aa1d822963f492f49c4fc8e57a14e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188944, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "cb32e9c7e2a834b4c6478d6f38f7cd8d16844f41975d3ddcb02e1753ad961c15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 224528, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "068fdd2789f1c3424c033900e089138dc8c652bcc0d3bc75e3fa0417a459f811", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 214672, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "36e169f5e92a763646f391ba84f324ec5935883f4728dc590dcf758dd2737512", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 224912, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "1fef81f8bb012ce1811b3b804e12f8499825ee876a43a5b588b3c636741b21a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204304, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "d47210e92fa07c9366250f422858b3ddad20a15f70bb2d7dc82b05547450d857", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189072, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "82da0f902844d672326a63b8c9e0d1f05b3b2a2a6432af8ce6787d2af17a0f30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204432, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "f739a8113ea81fef65deb7224fc9a2809b0b5b94118fe77367a10de061c40db6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193040, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "a8933896c232546c12c166b5537394799bc5ca23aa9f6c255553c9213445e1ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193168, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "248f749bbcb38337c13c2e2ab5036d8d85ece924fc5d8adf263bb1a7d11e48ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196880, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s3_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "a138b9cc4bc6f1e2c697ebeaa5b1d353c814d189de59df08ab3dbef1cdaf4c58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 224528, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "9a0ff6cefbc23c501d7fce5ec1fdc754b2f20a0c42527d2e192259ae2f5c73b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188944, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "37e6a624c935546a137cebecbab7a8e19269afce04b6e8584c59c71c22d0d629", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 224912, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "2826fb934845f8a360e21ec76ccf5a91f140acc2002a3ff53bb9c03373a3e4d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s3_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "576dbc13f9b07eda7efbe93d3fb5c933524f5ae7ed26367fa72fc9e04b332ec5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189072, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "b3f52ac568170b04e388368f1040d04704b248326eb7b3ce44d77eef01683c93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204304, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "13dbf59c98bf50c384f130d8dab3ff78b0e62bbf54ac099a7f9b4ce9e9fdb117", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204432, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "ef40a3b64310c868c5fca8c86f917c63a65f7703f0091ab9d001bf669677cb2c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193040, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "13a3a2bfd0f2afd3f30e2ce4558f847524cd94cc5beb4af2c2d9cc248f17c9c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "0e9ce0c8c839c5e3582343eeb0e6dd1d79228adb22120a48bbb9be4cd6b6b37f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222480, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "e341d2b81912cc95c4aecf22d98c9bd8bc11620b7928d4b76c61e8e602da2a68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193168, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "5c922315bdea043f794cc2b7208e240aab664d04232ab9f624000333d71b3d5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "7eec254fcfa4c26bd64750ef44a04640520ddd688fabe502836dfe01957f6116", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "7e890d71e6407c55e38174711fa7a7a15ccf6f7bf0d2f34c6fc61deb52436097", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187024, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "9b13ee8a4f7b4553a5da0e07e61c730142ffe0084449345c35669196c0ff49ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223760, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "75e46365500a33086a50caf3f415b8db8443b259d3554c09266ac096d64e754a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 196880, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s3_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "7b1c0cda0a42aa6768f9067d0b8e4bc8504ab6956329fa6b6193864f0122317b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 199312, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "089b383c7344beb3ab8aa08f4d7ef8adad0b3e26ad04ace3ac2ebed5549ec272", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223888, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "9043843d6d9e5be249a3dba90726db8c01c14fd354f7707c70d999def33a3cf4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "648f56628a9c52f540d8beda69774036594152dcec581a0256d56c293065feb9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222480, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "47dca919d997433b3bccbe88894f69bda1a7c5ee430d6753be5a3607b2596349", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "fe68301144ecc147f442c0c7061e4d5698d9aa44bc6a0d04a70e44b564336796", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "4b9ee3be10168316f2370c382a0f1a6e820c54932dada10c1170e8462ef99659", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187024, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "cd8f86179231c3cd474951ee32976ea219834131980e6c38b1e894e6cbf5901d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s3_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "d08d1fafa3c8fcf29a95971b268b5d4089f167341fda5c9617735e16fb1ffd38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "774578aa759d4809c398ace1fb828be3a34e2c6d1f1c2042173cee03e2fb87fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "47f705ae2bbedf1af298f1150065903efc55b111ce77a1b293b9bc1a6fff7765", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223760, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "caa814bca12fd672dbc3afde6eef984d4ff000f552cd2f2ef85dd6e8db966c91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 199312, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "11d53e245d0bbee6b51c4c2d608e4413aefdbfa9906fd7851e3979706935bba2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "1491538914cfad0ec8f782f032dcb42d3a38e29154210fedcbe391036e990a65", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223888, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "a1f7163aff69b426e0569fff0e674dcabf990639678c13bdcc5ee9c731f0a2a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "d22c58eea53d5507fda257ab1cbeb92e88011570d0a10b86f3f2b4f26e9a8151", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a8b906a7ab6185076d8f21599be350d7fad756fffd4bc5d5573e29a42c4ca042", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "075b22c57d5cf26503ad03eec0f50e8505ee774a400d32193ce2d004f50683f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5668758fd09822537c408746f049147190ee6190ed0e89b729f2241112927ff7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f9574cb10902fcae9f6c65724fa6527faf290d1ef5b54070bd6ab02adedcec8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "2e13be7f152dfdf399fe7926d53e4baead49bff399f5be73f67f9ae64831c927", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6c5643341ece2bcceea5ad380d481bab2310f201d22ecb64c1c5d9bd31c9247e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "7d5263c8fc090c25167b84f885c25387d3161a07af31d45be7f37c68a734caf3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a2e090ac5594480767dca37e592da2091b4fe1196c751d5ac3b8cfc6cdbced25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a8fb75953ad4fc8c35edab692108deec16baf429153ed3c559ca572756c9ef6f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "46f9a9895e080e58bb2965dac09e819a3f75b242f37860feb2ac9fb463e51942", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "065e10805483fd2ed53463d19a8a9b8db337ac14323f2709a59e3f48976e38b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "91fcf76fbeaacb0a211d70183a542d2902a1a53941b48f3fd9efc1c54bc0b03b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a7a0556610a5e08f186215f5998d8f7a9dd1d4f9f736f76679f1538cb5d1b780", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "269506e389c4820912c3edc067143f1f6549d0a5b09810494d6c6a5913ec06b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "9e6f2c1bc984b53851162d116ebb2bc08f42eb307c03e4511a7a5935a330c049", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "67646227eb6ba9b5e4da9b7c6ec0e506f025e2de499d78ecc043a424a3987e5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "729be48561c948476ee6cadf8f09c987b86962cbce8832b295a85d56fbfeb337", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "5776caa750a6380fa3f6a1da15440a4078fad280632abfb3e34d59639cfeb492", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "99ae7074b29e817c2ade99c1dad678147af6e28483cca11da203bb3b882af885", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "4af8cbe1602d1ea74cbdf6725a97d9dda4ff96d39bcd9f0baaeca2bec26316b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "921586a125c8549d2ad5d10273b5f775c866ea0f9620a4b264e77c6993e46e26", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "2129a8a33aa98b00f5923c9935537531befc91a8cb6ec660fcb61e1dbc6f2d51", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "98c465fde859e2752d887bc4f9d7ddda43d99e3f7ad84e9db9d56ed45eaa952c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "87aa85683bc5fca75f4ed665b5a4e0ffb8080dee50d9525b5d26c18362a01cf8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "acdd6087368664b83bae9c1eff5e3016ff323eca05732358bc04cd88155a3448", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "acf964497f7512b9fdca67700a5c942cfc66b7d92cc7867423f896a39decb633", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "f89f111de64e2c07cdb3166b394f3889002ce12a1e8580aa45925b7e816516de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "11e582de959ab92f955681ba3ec8729e3174d3126074a5a33b63c5a201654a17", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "3115a3cb4fe0a0b8211730441fdbd6d9d87758317f1ec682c4ff1b6784bdbc4f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "41f4e1347c247dc81a140ed1a9580ed3a6699e2a8eeccdf570fa27cd38011331", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "e99e247c1e30e8bba765c6336753a53ace37f2853b78a0463da3d1dcd78e2678", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "a992581b6c9b6a4da4bb44694f4b040fe7745b9f9695db61753c26ddaa9ffcad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "c4c9874b0768f35ccbf2badf8743e25d08773ffa92d7169c69cac7c925b16153", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "ab061c1503a4ff8b442271b0ad5f52543e8525369e1b6db93ecac0a59e688a07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "1a133755ac29226cd6f040e9b3b01a6ee426ff7fafc2f4a63b70cb7c11f68117", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "37bba9827900690a5d71f969a388888afe62b5807d5c793d9638d2c83d6cb8c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "3c204722e01e5ea6fbb918b8312df0adc5095c414a8d12e068ec485a9f4d8d55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "2e9ab1a357719a48f47eeaae413e5d3e5ea9fa4148c24adfce28e546f98517fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "d1e2e670d83525620805fd33788596523d173a6fe3197b2e6a476791f58da237", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5509683365bdc8621aa10fade63aec3bbab9f57eb8e0e3d8fffb91734d6ebcc1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "75e657c49c502073d87181a7159414db6b518865f48b2e624c97982d14ba566d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "85a840a682cbdf3040bfc308ff6934452b6357ec306d91f7fb347edc11a4e5c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "79afbbcb7931bc003505b114263d38a12a0fea9d73e59350b8585a4db42d8a4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "613c0ad51a2d938ce86e7f5510a431b11fc3943e0aa89b0ee77f4aba4b47a4c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "2d3d675b13c013f91c260f024b764545398671f7f42538f601d2b8f723b7320a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "dae013ff0f6c7bf9a0bc1378848fdd898c01b1ef16a61ea4f104eafefac69101", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c5e048c2f327e96eac860f1327732f3bed1d325dd70e6a4a5d119301511456b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "25cacc2c0d3d6aebf07290f48e385d877f316f8975376071cd295ade00423a1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "21f9f5ca67a6d790bdb2afd710ee82c1a386cd5a383eeb2bddfab1724f8f39dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "dd5c3d528dd7614faa10acef3cd2f91e02ebebefc278613163cd0d9013c3eca1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "228cc912385cb45381783e5d99620c55731906ef947d7a993c8525b686acd63f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "9e635ea1d629c00761339b670b08663630c66f88ded561fd7b6fbe2d621f7e1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ebbd1ada46bdd634c75f42985faf69315e7068785e0d998bc629bfc301cf5358", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "15bf1841bd36b37ee1bb7d47fa5a0fd3cd52e2cbebbd6e421e2e28c3dbf27fe0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "b6a501b68705346c1db25bdc87d952db10f9c4a2b6af5382481f1ef764949f04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f189647737fb3086a93f85a65b832d18387e3abae838c49b919f7fd2af8bd92d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "45d6f79104e4803efeb29a308df9e12d4c86ccfc7884b55b4f52611256a677fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "9f1e3c3a67f7b4c42da2130fa1339e2580895009b10b40af0e659bfff4ae20ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "86b009fbe832661277ac1f963216256841508ac015ca7a08e438489cb6b24413", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a6876578c6958bf97a022e547cbd4a3a6ad70f87b05ec3d540b19728da0cb028", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "20c0e315efd107105a6dfeca1820223fef9efcf4fae0de0b4323c094da7fb9b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "5e16404d1dbc4a01f648bc2a39f16caef64d231821b549ca9610e268f7f95c37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "dd7f60b5203d37db4efe8a70321ee7fa7f1bf2197e707623341e64ab6c9db466", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0fd454fc432f0693bc5fc2b4536811c6cfdb6ee3ed83c5c3e45934798fce54fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "56a3ffa0de1b46ef1c5003c2609ddf066fd6cd080e8d08fe44f9ebcec12c3c94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "7d571c2ffcd1f1ff2b6165091112f8ab3cfb57c7f419ca4b6c80ecb5bcdded20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0f4827dbf3b943fcbf75c531962c8a420c0df5e65180368252d43a4ff60528ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "3adc2d90bb3b379cb6f4db26cffa762eb5ddff3ce30f6e4c390d59f64b068433", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ae348174f9fc7db1841ad52808cd7195d8471b2cacc289bed9c1716cc339d69d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "917f73bd9ba69be8215e622f2919be42a232aca6f045c855d1a8b2c61d48d868", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "25ebe798a76121e07dd035b7768ad07a8a2b8e20b1f8f74801dfd5d80f6e783c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c30c906cf3180ca2a89c1206b3f4c935a15cb99d18b14759b9b9262f0be4993d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "0f146a77f5c9b5db867a3e609a0be57286581c39eddeaf6c165be87fcab13c8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "d5a17873d0eebb2d614f40796dbbb46c1a4b9afa255e1a0b8175eb7a26c68cee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "5e623bc02b88b2f32df1054af42f09e79746434ab3854dbe6cf5075e968858b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "490907c493d6845de0216710cd442a69b84e19819465346db2f60f24ba91b92b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "8e56011c7ace73f79b44f6b18b6a174be0f487e6a4ddf99ac236904c109ae985", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "1233bc8813ab9a2d83bfc7e6081c2d81e1730b5071789121057407b5f59c0770", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "ece702add92c37f10d11c2759ca47917d13b1d7db1d296dd0ce0344080a42522", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "39998cf94746f008b377bd8e825d7f0dc177b11009a61a36ffefac141a7c62e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "aae1bddca2f4d398fe6dc737737e22c808a57309d251ad24711d4e70ea051e25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "5d63ba8e176fd246a157e6f83eac03682515d86842e7ab4a92663a248fd773d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "679f2c0017209b32adb78e87966f42673b593adbb11821f2a123f5dade22c032", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "2691d697fcb4ce4454e6b4a4af6c5e82e031cb3d1ab8a387a7e5bdf55d86680c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "2d38c520ab327010c739da816cbba31379e50a7979ef4dc7868d2faeded7544d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "8472d09eb0fe46e0437faa15d0a170964772145333987f59a78699568f2c735a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "69bf9d7ee376de1f314f481c03fe978e5c4cbfba9196a3421b3f7c1c3320a9d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "a902fd5e595ffdc70b4485b0a586f7cfa52956150554cccb97ff386476389626", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "9bb5c60db7b6db361e92c0ed5946c575dee7d08d6ef54945e1c6653236d35a1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c1e6faac58a153df99e134e4340a16c64a22aa9acaac88c3d40412f6fb228a3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "5a66969365d83d1049dbf12a11ef04bccbe3726dea274689d58ad718dbd84504", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "80f9a3b9bb5ca37840f3602b6631983ba320a5ccba2b6d39b894fd2bd76dc27c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a94caf1cb941484fef1aea4aa0e641645f3b7c60a0eabec702cfd514e47cbdaf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "8a3671aa46012676e04ca52964647a945f9de551cccbfc0cfe58409f490356ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "e8de19dc3bf3b9df3482eb0427fdff07e68213711442859d0824641ab6024c79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "4152f9eabed493a628f9f58ed6cc16c74dd5ffabf1eb79ce769c47948a33aad8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "eadf88ecc6424eea37a0acdffbb3c5f90414a31e11af242fc31593423ad5ed7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "3166c38f02c637d364f04b485dae1f8bd0e8be315c1587f29b9eefb750c476d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "75231f6e2e9795d1dde544adec06189a3e3442098efea6c044c192dabc931c86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "2bf61060e2fb8ce9cff5b53b41b070f0026648a732c4c5c3bbb1eb8b1bb1523d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "f0d7391ea0ad92f6a470d698f49f6ad1d0366380fd67ddd17d281f2f53902b26", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "81e71ffbde4a1bed3728dbb8d943f3bcaf4388ffaf2a73d6e3dea9107473067c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "a42d81cf5568683a5b5f2eb21b2f26b1e0582dc7e3d094e2acf892be81192bcc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "725e6ff085f4b2e6bd3d74f9c26efb899a2dfcc1e2222f7dcaa958822f83c4b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "8d7e7c360125b851dbd54936432b42e4267e538cab9ccd114d4b03d518708306", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "209656dbab7ae97eebb3d29ef899136a83d6fc30a59fc0e5a4a76cbe2ca98aae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "aaefa7dd34d105373f17930b6b01036e0bcb5ccb05a7bb31782cbde7cfc13443", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "e34e97ae6b7da346e163f094b98d5b2abef462919ef9165d0cb1011084bd76b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "a003c6a1e410e46aa45e312a2374dc1c071bbc9e91aa6a5223d45414f265c3c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "40d58b383944abbccb93743067771fa7d71afb0f5f0e4e6744c48cfb43dfa063", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "0a794a05b0db790eead95aa6784e503224b8de44e61ed793d4c5b0dc91137e53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "8c35b8dc0fb4560e5e979b2db0ae8733648f84dc4326db0557aad49db4875f22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "7ef00b9a7ca360d75db278b6c3daad924cdffb9031efa056da859d7d49aeaef1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "bd9585a6dbf4ce7639532cc45b1ec4c5bf9940fb150dcb3edb74df8b2f96da9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "cfab9b1a512997c48b56fd298b57ef946c2be0bebb8c584626d475f45f233031", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "0f0a0657936024e6787dfc810b8ee604f5445be272950384fc88cce23749584a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "c02c0976bf8f4b881bb975972fb3e2d60080134246c9f382d5f598b5f0b893b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "2051b9d0a8b8d0c46c69c2027e42fce2278aa25814bfba60c68e86a7806211b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "2fa7809181ee6847b6ea58f775fd79272d9dd6c70e454fabcf94f42226a76400", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "80d8148c543eebc75ecef03a0086106b383a1111e3b109f932705aa180f79689", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "b437eaf7b6b73d319b47900e350d307d4c6535967e71f895b3c8300755c632d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "d0fae3e321ce98358e553d278b7e4d51ed63c038b56d7c399fcbb276b778e666", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "9f60d7c580c66424d359eed7d03f0f5bcf88418839f02ffe049c60e348990db4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "55a8c0e0050f702f46319dc0f727631056abde99d0e199b545c392459313bfd8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "4a47e35f8fabee0ab6d5b8a30060e4187f19391318bc4fd7a4a2898cf89cfd05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "3012e4458073c17b3e2de67dfee18657af5f48cb5da5adabc755dfb2a1542a5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "5aece33a76068450945a1acb47a4bfd25b6cac3d4bae42f3747ec641e6445785", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "a7f1d90a7089adae29077404d7f53b34cdafb62f962ee8be5ecd938acd526e88", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "3ddde4ca084649ea030a092aec5c69c8dd4938acf9472e1559e8e1e9944ec289", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "ab945c2034de3a30e9003ceb6188c7908ad95156d4f21d0821cba8aef164c0f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "67f90515c7f139b1ee52b2281511d996d964adbee592b9079e45015096961a04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "e9e30fcd3cc34dcdbb147731b00b40e07aeae1a5f417515fd422b2fdc1543c43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "37de1611cc436e77dcd5883491e66cfafe853c2430971c4db9993cee2997a65c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "e7ff3532634619a7fe53c0393d81da27fb93d764aa13579c3a46e971828a0d94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "87dc1bd5e690a2bc0ce9267eb1426e4c816a4ec1602f0b347a625ad54f7984cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "a8ba1cae73918bd248583adc37ea6322e833260daa79b705001bc9d67488616d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "3bc7885cbb98d2d6ff73d205d81d237fd03aefc70266afd0d98e802029832709", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "f286f8a1c48b2aec18863e5dcb42e54719871f3f318e18b0e76a09dabc2c81a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "3b66e40f07c9e0ee99d24f32c58168111b28f686aeac42ecc1b019d7d5b16446", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "69c28ce29ed9d0f8194b1d2f7f0743d35febd0dc926d7486a302446eb2155996", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "8d3026191148e4dfc4f762cf8aad13cadd71cd7d8837c5c21c9101f6cd67e7c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "25e7c6f552eb461b229c9045a40e20b6e6cac221bd98858b45c351e7d79f6ed2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "3aef6dbb18f3ddf1e006e74545cf56b30e990a0ee8af77a60155353a5ecd1f22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "a1bac4d6476e511b3bf26c4f48dd9fd2c261f48d455feac3a70b76c490b02330", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "37178d578f9f3448120e2d77543ffd12ff7cc0799b26b7a9f2b9eb50cf5f4be4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "1e49b7de996f370b23687d41aba8bae17498b8ced3cea725f7bb6cf053c30d38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "5e038f5587fb921fadc0ff719ee963891ece88fdd3628cc8b331a6c4fb3c3076", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "4bbe6e3b06a819ef15af51f6e61aa9f574ef9cf8d8e8934749516b84e96aca45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "17d944b31e6acb0ebd8946cb5c611f68df36aaacadad9e271306f3bdbfba68b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "d6c2cbe11aa807213192ab35a3e5aea99a3de7871d5048e3f04d3a3d55837691", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "3c84c00f919db45f21c45f65c7fddeef6febd4cee7b844f35255086e4937f240", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "d030790d27acce8273706adcb157e3a0dd952164b74ecccc9f71f331e48e55ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "a8048af461299214cc421625080306d6e59f729f878b718aa68ce448383591b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "0ae257b2e9712d0c5961e80896e42d83c2288c9c70da3b48a0136bde38c14fa0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "a03aa078ab4a62597f376840076d5b6784fa68c5927d3fd83a2ac41c814689b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "9b6f6b78d48cc8179b47da87fbee7ef3d0c8e47ac7e17ecb439dd579b17ca85b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "d78956e9b803be31f0620d5e61b06260a274ddc9165208c2a5fbfe5502492044", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "f57c4df68dd788756acfafaea2681d7c97a3c27df225b098ac605fa1a55f17a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "e84d8e483dc4d6eb36e0b6ff6d0ba8ec79c2c3158aceb765fb38081cf1ecb7d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "c832eca12931fd08a8c554cb9d528128a6094f43e768625c15f2b46b7d6db43b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "b535ac92a00ef5cfa9ceac474f76ffc8d3413e4883d1f54057520051a6238fb6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "47313585705ba563cd1b7e02cc64d59791c195d61307bbd9b299152528f17aa8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "8482b0ad07ae7ef4749a6d020e996091dc6bc1eda3b80546f96143dd4d5d6812", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "1fc46c9f0f97f338932f72e52c973ed5b8b383f9f5eba59b580ed7c151d49d21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "1284be258023c0e1f36d8a0e86b7f001183009245229217bb097db90fd19f522", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "5fc7371c0f3ae8a27483e2c35228b1f26ae4294db430cef22ef6d218cb9ed0d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "58dad1aea952d4e46ad3d8e0944a548ac18c5c568759b45d014770e0bbf3cd4a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "47f0600dfcb0f4011d0ea2c952771adac4452f464b8f62f63a26ba6c2c33216f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "dd1352e532e19127feeacb053afcc000d5644de5455dec23d89cbbba87eb5489", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "b43563e5a4f585445c469be6222c618aaf47b0d2636c87bdef11036ca2807bc4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "ffa717afbdb4f5a64f6b53768a618747a502c65505c5e0e1e9a9d28759f46513", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "640c1bfdc95e814c6311cb0555dc2522c2d78e68407590b68c10bb183e3743aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "336849de7f2aa532085d619dd34e9b493c5f67fe17150d228359e4074760883c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "7fdebbb90c217f6d15e6191bf5c5cfbfa404501e1392651a533b479208cf42f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "0b91018cb6c83898b3dff45746e86b5f1ac638acc97a0bcdf72b638455d3b5fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "b3d51a050f8f55617495b5aed5da9fc44992873fe21f4f62ce0c6ba70879156f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "9f4c4f871469fecec82d8b02b5f5672cda94bd588e14f99d912dbeb320cfd975", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "3ecf2fd4e1efe7736c2a10a239a251228cf8c555aece2ba567b212e9e85d1ef9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "a09f3fe9a0ce0d7f0859917c98f27b3bf0bf412496a4083f79275464c2340b8a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "4043f37f94c7058be9195d6ab495b61fcca2d315b13e0caaaba68135dee38783", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "fba1e9a57c757e302726ba5548d104f2a179295ad7f19dc55cc8bcf9b9bbb291", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "209c1b0461f9670470beb9038e2e993e6a30c8aa577c014c422835ffaf02c023", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "d2d128c21fe6ce38570fe1d8b36a9f98ac13d01fa88e30019a6faf91645a75c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "f7fc7b6ed8d3fde7ff795f5de2eaaecad9fc8d69156cb90dc1e13d0634ecc37d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "60916c5c6b2e0f10b196ea2ad5aae0231addf44088162dd785f6e1a186e55eed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "f5685ca1b05ef12369d80dcb61c0a93c780ffa5ccb13563abf34cbeef4ee02e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "4b33dab4eae3d8c2e182f10d7bf719bdae56bb0ec3bc521bc2691d6e9eb64e67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "9bd3e5243b99fe253fcbfa8badf4be4434a955f29f0bc52ba3862259596bcf9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "6f45f54ce003c3303050c2379e82a0bb6b1977d48f615bfc9f9d6fb8703a852e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "36eb8746d7319171db57b8f1fd4359693fd8cf1629b9c20504068530f6937740", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "1aa4724b8a775449746d9b0288678140238221a20c8a1e033a0ae49b6e54c1b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "6ba6adfc0b8eddb5459eef81bb77f4250472d6d17b1d218cfc1bc3f0c7d71ee1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "b9df1c1a409bf5879727ef1186da7a1000223c7e119fc2d36ac790f240415aeb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "ee60e55e6d19b30e055c0eb0fbbac64bf3a64cc5fb4e18f1128c760a15549104", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "de16e25f71a39710ce8f92c42823e2d41e6372cb09a97839420f4e861a387763", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "96f8bf67d557484b4175b5294c6a66d29f1b516b46b3fb8f762aad7bf5a6534a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "b6fdfdd024f9d4c50f96306ee00aaabbf07a924994cf4e2bfa9ab04937f3d40d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "c5d1ea0b9fdabc42d84a3015f854ea7171ade527a6f061327a7133f008475b37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "3cb2016070d06dd350879093a72e4fa63c7de64fb0e8bb8193009e847981be21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "25a06de5f52cc54b79bf832fab04f2fd32f88b3917e919e58e37399a9736fe23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "c7d7cefbb0e55a8c282356cbd408a433035041d6503e75d531083083595e6095", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "07dd08a0401b2e03b36d6dde72582e6423a423bfe10072147003d69612a3d5cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "55a02411b7813f01c43f52f1292a898a548b329defac23da04985fdf6f9c477c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "06ce5aeac773119240966ee2efcab085f38ba79b9ab547fa0795d189a2f92e36", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "206cfe2bdaf6163133d1b5696441f0a32c27104c98080ab8267a4fbb8ea2308f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "fd0099937169ec242d1ab1dd685397f8417eb965720a2502be6d0727f9fa6046", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "3c01fc5d17392a3af3e3b880908b7977585e42e2bd72a8c64b0044d4530d6580", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "084140215f6052592923786b97f0aede5e5399a489b758391ab714dbd6b413aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "2d9cabae87d9aa446314fab6e3ae96b6d303fde2bf646fa71b632e43ff6a2a37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "26d7163f367fc273ddf2f18306d46714ab3e16508cb4aee6c73e4e14fd0bac8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "0923873bea321a21282e2456ac797c4b6708c21f7ed80fabaf584b7c3d0dc696", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "1d3124c485572e597ffb55dec66fa14b6ccc1c30f8644a1e2c13e37708c16b02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "3e3960eb3a710c6e4727d4bd23a4beeb142df76a49fc80be0af2efb8b13ee0d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "104a6eaf33960ef4d0c0b3abd42106ff25e910e1f7270289361a783eb41e97c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "891c917cf3f87d810daeefc66060ec23816780cebfc20967c93fa8f676b4f7a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "c8363439e3a9f34566f0d82a537fc5dc190ac82f9a5eb27c1fa7a48a16bf72f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "77dcd98ac4d219255f9aec93c1515043af2a053a0327c14f2838ab8104f9f053", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "f0fb000df6597ccba1a011ba24630847dd084c7c4b0baa2d781bcc086382c485", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "059e4e2bbf8f15ddc32b3030eefd03a87f589142b6d45918cad7fa1d88825024", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "a7a809c647a0b107fde6a7e5b8079e4a96e1c7fd26e11f930b6ced1360517da6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "b979edd00833eb264355aa2b4432228869494732d7d5c05bb76076500dcf06c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "89ea1936757b03ef2690e3d5c7bc1d68f9cebd4c3000be61805508f83a9dd078", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "01c73f1c23b3023617a2dc35a8cb4a355e42c700c5663f9e60b20e66aaba1383", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "fef7e190607eb9dc344c4407c447fea3a752df0e77c5489648cdca42a4c5c7a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "d106d5e8c44d9d949554bbdf402561dff25f67c6be914a824b2d44c2dd1ebbbf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "afe45e3f54758fa63d29a51bff43a5ffbc1e1a4a4aa14478a6765d327e6763bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "1cbe3841fdad0be959de62fb2f6579ecb773567e1219f99f7996396ad86588ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "41a4a02efca50fff564cf090bd64caa271eaa17e23d5750340b53c3fdb7a94c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "343f4370c28f083df622e04f4946d0d756520ddc6fc246f25ba7e0275fb94568", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "a35e771e54ab97617b2ce23fa00720556745aadcec01a3cf6155ff192a7feb6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "f1affb73fe2f719360cc7a30b99fdd466720593dca6094806a114dbf3d22f2bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "4e7d74ba162f4771e1757ab223227b60b31f7decb1fb717fdbabcdc340aa463c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "5a029e752aa95b617dcbfc4980e48fe6b35b688c5484706f69f09962310ad2cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "9cf208f6a0bd3e21ae4af745fdf0b6ab3fd8e565c50e763138c86c5cae751369", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "6905a5dfd176c2c11ae5906969e5bea10009d5898f362593b1d1747ec047f000", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "5fa845013ac2209de591345f467ec8d776624b7637a943be6a8ab16da3deea9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "b4141a599e97b68c0d40f6c48f2ea1a1dabf6da308b2054ad6540d18c7338eff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "a7b56e0f7693299995106ed746544b17bd8052f2634aeca588070b1c0df46bed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "9169e9124e2af2c7952ab3dc5880cf2d9a7869e743bc31209df8238fdb735d12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "34142a3058617fbdd5487849bba321dedb803fee465b620de14072a9face34f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "16a9592372bd79b9b732113edc98d440a7c9545a344ca091d54889ebd4631cdb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "b9c2907b28bff9fb06acd489f256ad24f4b1b7a21baf526a39e0ad517a4c2e61", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "9348fd3edfb55c6d09efca963bf0660ec9cd84021b9bf368e0c468eee20c403c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "0f49a0fb6c0dfe08ec1ae8c1254ead52eb66274bf3a63230d4789f09d2885e1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "c0c356daee5ba1b8978249b85c36e21da3661efcfb7388c97747f63d2a84b7a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "da81cb4546593f242e8851caa9ff3e01a552da8d3a36a5e8732fd5035c440a8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "577068af0376c19e7da08b5aa338888ca4a872734114caee0c548d1e5fd02fbe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "1a05e7e0797fed481d01862231d78e5909970e7e00458d464cb059b0428f3dae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "5586b4ab57212f84afbb273bbcd04caf3bd438ee07287ee60ba42e71de3645d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "6075ccb9db745af45df97b08d9f93bbdba3cfd5e9dd18a5e28419de5279f7dc5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "3bd149a2eb5355c523adcb66b9d806ca1b79607304265359cf6dd6c5969f4321", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "468547174f0cbe457e8b17ad67ab1a88d11185e399cca6fc6399ee4cf6585451", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "9db74c431eb524817cd896911126315b7e6979f3f828e7acc616d3d564a7ea91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "481856bb032145d5ec2bfc928036f2bb9c008e8fcc221be33d821a438908975c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "93a8b41ad33342248421de1b70efd4e23c69c0d80eec36be6883f4b68e77b59b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "f1f4211390faf1f7874937f604afd4af65322069d666a13aaa8daf5b058558e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "fc237651d2a18aecda3ceffed1cfa26152d2905a01d7d04eb36efa017f50f04b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "6f1a858d853aa5f75c9785bc2e6581d829c03a7f55d3260e354a0724620b32ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "084efb1b2a7a097d002cf5c74866f370fd4763f308c28c8358e08b8256d903c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "a893ca45bf476287ca3121a2590e04038e253ef74e97fe0052c99134b2761b49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "170350db6c31fedc39766a9f77bf7683288a785918f92d72659e7a3cf1631619", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "06f1760a2617b223f01a44d6a9b6e3afa42c350ad83e5e1a1581a4cbfa932901", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "9e8abd22253db20c8f670e8761809375e325b13b67215506f3f5ccb107f24669", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "95322527d33ae5e622b7b131298723a4e2b5cd9a56b70f3dd3270eb33a5bb13c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "80342b0b69b451078c1627f9f80482c2391acdbe6747bb4a6d61e7ea5419ca55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "5b665e5aaa1cec79366a1760df4c7eff854cf9f713df1c36f5b5990351eee5b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "973ec48415c14b0b87e1f3a07d00428b418e1b186c825a58f27049d5e5b2f8fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "3baffad085207bc609f1da4de7e567f2b7bf06b949134acd8caa3b8130268d14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "332fe1760e5a9f94e939e73fcfc8f0029ab32141552999691cfd10ca100bef50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "a58c943f2090f88320417ec88a7d0543dd1ccfccbfc89a65b9c8f253d7784fd2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "26b2f2cdf289aef0b0b1aa0dfd2b0b3e98948c354f0bef65dda07fa8fe702664", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "b88145af99b2c01271549682de5bd49799f9add3c923d4d3a9d3ab4db650249e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "85036a48343f44491afd3893f4cae09da14de87c078e365670f0215f865c31ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "f66e9cdeda54e1df74be569ff618f84820301327358cf952e224759a54876af1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "7e52153532a4a7096a86dca3e12d15fe2a1633c634d0330d606f71b74192955a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "b06ccb1ec8c03018db11257109425c379e12af36168040c294fe5395d724079b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "e9d08ae759d59eb8d11126039800158d945ce6992479d97fe193e70bc2277925", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "fb72b9e3e633814b6f837924db97a5a8d20b3d10366e12e871256b08eb5971b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "89d15e6661205669db51bd2eb9cc6d4f5e06fcc3b2bf9c3eaffd36973ad13a38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "8f65ad80ec01fdb38263eba359d1c4500059e4c4b6c6bb50c2c87c0ecf1b2329", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "9e25aa795fe23a20980cd0a4c4925f2e713f03269726b6251a2cd3fdcdf3bb65", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "cc0882435107411683977448c98d2e94bb51e8160f02b21dc737adc428ed56d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "14ce1ed76bdbea288b10905315b81a30c83d21218c1e6ffb494f5831f4267dd9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "101be6abd8ba10aac947cda0dbd019c1ee421a258a80f42178f96ea9d63e868a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "41dbe7e1ae572a90a5be5f5795ce7d79fb5d28cb2280f6a8853658543b6140fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "e35a2fea29bc0bf53c371a0f3df53ec6fd8bec5ee7c9be0b5027fe542337f9af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "878bb3d6f79bc9d8c61dbc3fb3b2e7f5bc9661aa28381fc81566ff15dc1e9064", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "4e18abf5004553b1e701cf36d5f64cfe9707dea33c0afcc18db9490478571dde", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "0a1367b38d37ad481146583e1e51ff45031837a374e1aa3f866668eba77c9d5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "ed027846800d34458e92785429599ecec93583c7e6d37175abcbeabdae1db29b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "f511ab28921c328eb039fa7ff00de9188579804bdfce993f6d18135ce8a6ee52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "fa1b707cfc870b21e103f3f8974557d20c3853553bdf678433fbb0207e1e839f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "b1f059b15facf9fae2e9b0e273a5534ebef54f76f7fcec2756ff70d4bbad0811", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "9d2a61a11d6653be0ec6483b1a94d58af08d4224c68b5a87560240a7ea43e43b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "dd3a6b184529c13af70500c56f20b06e72cf9686b6002817e7cb040c6aed977b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "fcdc64fffe78f23d78831c68a8651e8dbcb1a917aa9b445e9f67876779204e00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "cdfd90f82d103d4190a4e13dcff5932e47620d9876568322e14189a2cfd368bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "325e940afc1f7e9a92db802331e25e4091f13eb5899dc271ccae90a07b020e40", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "c7a557d2b13bcec53057f7cfd2df0e6e4145718e557e97a3700ff19b56dae686", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "8626c8e6bef635b7e7e8b70e61306b1f3a0105a87874b4d6ec39ea51fe268a91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "a5624185ea3f8b23ee3fcdbfe92911fccec47f721a158d3d530e19d0d86215aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "f8e8acddce22553cc971039008bf16e1ade98fca2a0087cf35afbc9df5f832c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "d12c97c3028bb2dc56d762949292e7ca35620bf6e87ce00af6f69e494cdd1093", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "e25c3afc12070d1a0234fea03f28bd13a428dca585ef112de0e4c8c829ff87ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "7b855b052241d1fe694021060e63ac14091a02bd2ee73f89ac31154570dc9e44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "6b9293d3e0c641d446d8fa9615abef196a8b0f0561a44440883e6ac085816cf4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "112076c065aca48867288e2c4179445591d287583baa8d423e3a5aa6a10404ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "a3fb4116473a878af81181376515022b580e9977b90eaba0f1e5c4671b585121", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "6e6af31e06098426f0eee8a1417ef28dbbedee3443b6c68f965cd69f2a2c77a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "9931593bb3dbdf0519095a4458237548e023a65837672c9bc5000c57f444e52e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "c7819a66760562f1a442626d818a681a6e16cfa97154c4643cfb1a7e87dd5304", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "13c23d517b6526bc3798a9a13334925ab3909ab3ccf4ca95248fcbb76c5b3817", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "65513860671bd39e5ec393034da67da320f1c725a643a65a1665ee54e5b6a94a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "2b788c8f0185533d79839c0ee133451e3620528be2a6f09900f16c56ef148ad0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "3137d82da4a578bc85b0dc2751b78c7eea8e0e32e59d22fdde0935551d01933d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "b62b31ec1631dd21aa5159d26251eb2c21e2a6ac5eba5425becd85e64552b9e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "4c70a8771c009a667969ae86c83f65be115894792be78abccbb0123651b0e425", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "1025198eaef98e89fb7366e3b50265b53da887fd82de46656a90eb571b93784a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "aa85ed4056bd95b3f8b47c55047b72e9ec017fbd90981d7ec01ff5ac9ea265ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "a401af477f84039c338d573f41374226d5cbb6937c62af2621bb92be7baa375d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "5d437828f250dd6b87d3c97835466ee856838a9e3b078a876357efac3cbcf029", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "115f5598f0059ebde90c30a245e9caefdbbd8a6bdc4f6c60e25cc9199b7c2586", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "ec60d4ccb1c63a04c60084a8ae8334bcdc2747c3c6b85a90fece8e0e75dbd746", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "e1634619eb9ebdef06abd63c3656a7ec080d41a6452a28e5d3826867456be000", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "749b91c65e641fd773fe2d27f5feb54156d78901d1ae2714380ce21bb3185a12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "1d114f757618f871bc25ae37f1003dc3d61aacfac35aa1663fedef9c2c5f1ea4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "d25316f03f4fff4d123b2d9d7d4cbd7eecfc92320926fc1e3db0798155fc56be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "f3568e7e65171a9c7a92e27a72627e718ad9faf3009cc2bdba18558de707a300", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "c80c8c5627eed5a4577241940614b0ff077472e977bc3eebb95d8fc6692ac82e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "c8200dd67825a76b3a591aeb6f560f81823aa7ae304deee2da295aa7b4ca2e9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "cc285acc3edda76adac64175650c6c6a99805e92d3734a701d8d58639f983345", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "29b8c3158c9dfaeca8d6317f3fb6bca83c4f494790675da27fe71666e3c7171c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "e51dfc7fc26bc94758634b4aca7acb4aa5ca8f0e85972cc35b27e5ab4825b7ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "36ef6e29410efb0622d0c8d9589fd45cf6f60b72592e8e520980466a32430e1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "629d1ebd34c03df110673f6f321764e74955011f260d6ff23111b6dae2152f9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "60932e879d81b23d124187aa7e63e2b62f37ec9ebf03f7aa3cdcc0ea29d3359d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "90e6c0adef410ddcd1149c7099b3b9a361b4d1b3429c3306469a33e2269aad09", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "670e1c297df4544851defe86a6144deba52c749c735d41d72afe727585e36f21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "9c70d334b9d3c410a0c60a317cd5c5196bd87990586061ee1dfbba9adbbab1ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "7c185eb24dd07b2ef2a3f05bd995a42a3cd56100bbf8de2836d05908beaa506e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "2a63b6daeba84abbc35aa98af846c30fc582a0b0866875f3fa996683598f8d34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "aa3b63b017cf80f00ee213549a524a3b348218b1f9833f8cdbc0837091dbc080", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "6623bfc724b62e95dbb093209e5bcb114f8e58fff2f6742cf1e8e5d19a118409", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "d070417908be8c8dd45d12af454cfdc21707c9d768f10bf701c422ac01b5b66b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "6de6684c0595cc5c41ba8999de3878250c89cb8c4a25af35f436a7f293783e54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "d7ffead4b7870569f6d346053621e757d5d24a24e8b5c8c12ee5c65f1b00caa5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "3916405ab0ac440a2602cd612a75c2099a5335d4abe3c39456b00248cfd82ee8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "93a6c81dec85ed0724b2f0ef964984c12792f197da4e37f83329cf08f0bf25f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "f5d5a1f608a7ca040d12a0ff201111587f28c27877500d4542afc93709a3e4e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "7ed33b76be9b6c1a5a936f537f600240f493b84cb75ec310eb24918066954d6d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "9197c5368e506d4692f423a2bab45b67abcc4a90ff01f852c61c29b09623f92e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "12f946542505f756a268e95ea9cc9aac203333f0ef7192e57ed329533d8bc819", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "c1f5237b8b61d6fe271ad1bb2ef873619c4ba668f0ea72c4e7c706eab84af2f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "55c03372576869f7b8a87d91983046f115456955c85b0ac14d6016e549c77645", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "f4d05bfe9f6557de586434fcb44e025172b8bad3461f0a6ff284de9c157df5b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "c7c7f9126a8ec03f77d05bc6742890d9d0af10135d5ee03840ddb251c0ab6569", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "9dd57de8148b065e18f88257dccd5673dcbff5f26b5fd26706d94fd50148e093", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "7bf1864063dd3021b46fa7dbb2dae6f56e6c0449c82445150ac2c9a41c457984", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "c7cdaa9bfe46f2264dda452472f4bc7ce640e3db5a3129a510057b8bfc22b1d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "63fe94a8e628d858ee4239cee4c5a1ddf8e957af4090317fff3ae2bfd6cbc2b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "fb499179af893b0f173cddd649907a69c1951e2efde2df05eafaa166bd8ebf14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "1674905fc0a3acc046c05862db6cdeac19d03d005a1eac2babda4ff7e321c176", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "a5fd08c1ac397509f5f805a9507d57e505474168e05dcfef92950b394bfaa05a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "2d651abbb879a61d8348d790245cd8882686af05d537411ad361df45d7c49d83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "8b17a8f474f8eb6982ba19942004c30ac50a3869e8643882ceba3e1f746fb38a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "18ec71ffa9412a456aebc781a201198fadd1d39a80d2aae81d401b8986d911a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f61d7661dae91eca125acdb841366d2263ec0b56163c557246fa1efa16209145", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "c5bbb8e7b729bf51773f6a4257d3abbe323e9fd33751f98658277ef0d0bc6beb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "049c2dd146b73eef0e67dde6a4c1c319b06d906480be88d003085ce45f716a38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "94bb7678bcb4cfab268923da699dc33c4769eb88f5a7ead7db8d3aaaf026ea17", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "15c00d918799f40cd5b541fc00eec30b7afc0f9618411ce550e0b88804bbd431", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "7cf8208110e008c9d3b0645f9bbe4a4651fae5b7b0cc322ce8e0ec7813a43abe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "adaae5b6359946a992ebfecbd43abd9a7cd1ee9e01157a0845d0c942eaa09803", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "44d09dfffe3f5bb95584aeb97645cc0f83bca5743e10ebbbd623958ca2fa1f34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "7a5f3aafd71bb38ad0ed474beff63dba684c55d3a3dc7b424af0850e713d1f99", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "cec443b73a8ef779dfaead891a0846c6b9701a329003ace4922835a630b0b525", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0e7185a6088af2f8c967d696d99c5823c4ca54c068f626b8d36c6057e0bae425", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "61b2b6e8064d7e61cf0b70539472c3f692305cc92dd6a9e0c00b6dc21f8f050b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "624f0eacd5f7f0ca735029c25f477cfb1438d3b1795504ea1a10bcb6bcff651a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "8ae2f334d77f4fe926f2d3fb4c7b3dc7abfd0cc1dd290ba5a6784ed644fa4cfb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "81c484b9787bda6a8ced81c979056179c0129bc318b35b5eceb9a8d77eea003a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "2229acbf9f027a4bd2d09d868c60d19bd9f1c17ac7a724835306dcfb1aa6c105", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "58a3356f1142cace9a6634a2bdeeb7f65bccb42d4406f55d125f2d7dad766dca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "e1f96d1ffefd33f426d7d1f8e1c5cd294116be73f0fd8190ebd52bd35d5b90a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "24d04ac4a6bc5377bc746b226552644e18ac1f07c50559a805003acfc800ea00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "4e6a40c360b7789cd15ce0a0a5e1bff5b98b2553fefa24cfbeea59a64f16276f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d2101ccd6f8e47a89f5aed0903c1fcb6405025fefb9e823366240078f8062f46", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d8ffaf841da372a01f4ac1ed5643770f52b1365e9a3a38227148ecf5dc662780", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a8bb60b30487a1fd5bdedd621bc8ba1bbde40c1cbc387d0c7d6a5505f617890f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "67b8e4c47cb0d9d5b8fdfb1f4423b53c32884da11fd30e3831e52c6d3bbf4f59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "9a8cdb1fbf89df22cbbcdf0f10dc453cb856e59906ef0c13189210f302a4055c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "fd1772d645994ac04a36fbabbb222758808150bb66bd9839dd490ab63e2cde7b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "af6e8bc75513771f5ffc80bb3e73cdb7f822650237329fc3405a3d9cc2a771ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6888c513b0eb2564e9cd017e740f79411d9b265e97edcb4df7c321719b598ce1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "6a321597d2a7a2ce6a11fc630c0c5561cfb37644598acb5e1ac99931653169c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "0443387e48fa0e23a999369559c51744aa633f69d36786d95dd82299ccb3631f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "97829270b029750221e2c82e91f59f1e0e53055a2c50012b82e3b376af624764", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "e05771f35412ac963555679b6f6e0ad0aae3e278b0f0c55db1112608cbfff204", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "5e67b5acbe54589c11c0c4c02685dbbb74a6251fdc1349bd75b10dcf168e5417", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "3ff0897a3599ad814664ad7ad254da14adcca5cf1442db444181ae87df153641", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "9873d7889ba4196ce6a656eb4b28d53f739fdf07745de14bc2d743b75fd0c779", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "7b4bd3b05d90dd65edef42809afa7bcaaf3da4d74734cb88e759a0c45087a878", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "4b3bff712ec98f953ea8aa19cd7ba8ce810fcf4f3e671c2033be645c31b40608", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "1345bb004f4e7a0e363e3c6478c9634032cc1bd2ff5eeaa4d6638bf2691b9c97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "23e7394d480744c1fa42c41339b0b3ef57139ef154279d6eac192498fd1504f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "2656cc96955ea0470ee8c4a6c1830b8715701362e73965fe75ea8c76e2376f0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "060f3e7683716d88674ba346efda2e8ca5c7efca5c9eb5fe3ed136fb88e3f3e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "a31de71fba72af9aa89e5bd20499a50e2256da4c2d43bba4111e3ee3cb887c55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "9b4236248da1f2518fcb040351f727c45f98cc6320bc49e6e676296f9bc9c46e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6a6506620d300f3efe120ad1a4b556414aa5ac130ba73189905ee1be3f86a9d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f81db6387b99be0acfc4e5b791444d9673872d24a4c217ab833b6ae8411a2e13", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "b54296e05274ef3969a4a8dd9975b8dd687da3c6d330b3ae0072f2f29b3cdbc4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "b1f547a3e3106dfd94030eb9764bccf8b0a594022cf0a9218f5f5260ebbd7a1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "57e789d63c974c1f309e2ef6abc085ac8e7486a687d8a223cab56b617679eadf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "2cd43af380da5670c10406a209f74b850327488978b439c40ed3efb9667280da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "78f51d006a4d78649e820589e4f5d46bc211ce7aeb0622df952eaf176e4d7ad2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "bb46c9151dbadd6c34d0fe60477153577f92bd7166f930104868b604594b827f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d8d50b956c301d4d76f180afe32233f6c25ec2074294b116c3289130b383d145", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6e77a19d7fb38ce57a67764b177f62677f814d67ebdc020ab976099d9469c87b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "1ef543cd10aa29cdea29a3b12f4343351303feb5267781901e45fb594d9500a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d9a209c2045b2f0ffd26a2a8dd0a3c2c32617b0f135c6b308e3362297b1f0250", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "bc354b5699ca6a92777638ac32b5c43387ce9a42bf0f9c90d73a74e68a164c5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "1df9fd43198995a5307da065959dc1d79c80c7fa137394a5fdd1bec2f36515fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d9bab20049866b173b56ce1650196ac4b95549fafdc646caeb202648c4b06103", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6d37e7530740e7f0660ee0616a5b7dda86d6bb5f90bd14009db06ae3c3c8af20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "c10e3ada70ea938245eae2f48285ffdd83d14f2d54547acc8b93a990c13cb042", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a02b2361b64c904fa8931bf1ec3846e413b1938a9084c34e1ffa47403952029f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "51e0800dd83f0b81cfe2875f69ef019a378e43ede3d8e79047c97b70fe9b915a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "18f0811eb6547a00cac10fd4f559632d85a7998e86992a8021ed77a277ee7db8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "129ff2fd9831f7e42cfabb069905b5893f12576c4c56e153c90f89600a9dfe55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "b04675c52d09935d166cf540758d4b047aceb789a1ffbcf742fbb4e7e7386d39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "75ce11832be3291883ac59e342ecb2022cb565330887f30949fb16c16c20fa18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "2405690a3b3dedb80cfe0c3ec92398992836c3d438e8303663caf8eb7ccdd530", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c27107734ea3cb06195d5dcbd17ae9339a467c83428fbdd835348d5373cc16b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0444daad2e78e10d9e535963ac56460ab584e7bb5dc154d64920e9b3600bf6e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f8a5bdf64bb17426b55d4002ed73a16e9f68e95436d112edbf0048efd5f65e1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6d9bea3f8b2ad84feeee77defbcd85e7d29416657684a893eeca13cc59e9b716", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "7a7805f47bd39cc815c2e2d192fc5f37d2070e71c0ae0184a0b68186cd1947fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "6ecbde0f4507079c80468bc0f9fb6478a8ae1444a50aa71d7edec225c1f06e4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c0ba133b88143ad85dafa3d0cb19aeac1bedbb41cc3d4e6bd8f20c57ce061af0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "88b7246db946ece3e79281be6273942c5078705f1d13757fad75a2fcaf4195b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e2b917e9dc0e4fdc5726ff2f5c12a3ca7ed9ca769d2e49c97721757524d2af98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "57a0cb68d1122a7ae4157cff101d8928582dc65b9024bdaf9e4c88a837dd2b69", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "13c27c508ac9f8b6bcb921333826e717d677ad0bde6266d588d125bce6177b7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "1ceb117c777e2a2ebada82514b27b49c82499292b9206c9579fe4112386be5b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "c03778a8ebd26563d55cb4ce6491392625622ec720b16956295b8fdba13845c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "6a3d4ff6d9accbdb206ac5bb22dfbd581955faea945a0537738809d827cc3ed1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "4acbaad65982e3b3e2fe0615045d18e1807024b58c84177e068a5bf867312e40", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "dca864d1a1d81f8c065d234aea1b36fecbf3441d26929f2814a905aedbed6b78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "7208b8bd73e1aa813fb3815634f50ac1ae1d8a93651bab72f2712830ece9fc6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "90df805b254625bd57f29ac1ba2e112243d7951e665a85b2e7a733e5dc0add18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e2220671f665911dc7cb480b83976318150cda9651de29917893eded731632d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "5cd153f71bcfe67019f3782c96ed10f035f422f6cc7e875561bfe149ad85e830", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0e65e6753f7ba0d058ee89e60c24b3f8548bc3e40d89982208e81e82bfe8da78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "94e910e1ad8d0325566c81501548d1ffaba7b9b96a2aad44f1e40c70c222d1d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "cc36bff7929f8dc7c43d2df9a98cb80c8e99b2ff35692751d7f283e91874780d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ae6ca53f39fe4677bcbb3ffea12d0428725306a134cc15ebd0e89ca39a878a00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ed7aa229db7240b84fb24bb59db5d91b8181d9c6fafdc2f506a247791c04bb3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "8948f9e9b978c7f058b9771781b90867f04af0ff27adfe236b38eb11df808dad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "7b0320c7715f0b8638128e1a53ed959342b937ee0513eedd928de510871c9d7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "4215968e40227ea71bb9dfce3bc064ed8833a3be30ac115410cc1a0cf15047ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "35df5c3e508c8ee2aaa5ec018ec9485077287ee0a065dab5948ad223d3918c93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "13c7662b96ad0998d2c66173fa5504c156c95a3cd90421601cf3f5e96c87c3e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "7735c05ad3828559dc2f94d6a77db1db94de12772ae2772e5197d603a8b90f03", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "2cd68276ff56e91b3613ff9d13ae459087efe9f1175e2d11f2d95faff67f0d1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "e90d26d08d9e55002c329d49f4756ff064487fd6f9fe07cdb61bed88432bc6f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "5f4ad7b1f4e65e6f254773c879ff6ec237990246f9eede97d2145d9053a11061", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "fdbff7d39078c88646a241e4f0bbe61397fd8897badd113fae7213c0d3ea730e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "dcd77e891cbb1a8f2a56c51855e50e37142149b9105983bef1dc49e37742a373", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "464b919c632f716874a33abfb0050a8ee7590517de91b947869ffee0862b1fa2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "92379f1d17ff55d15a8af319a6e6e93284ebd711596841d6f2839f8129c5df6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "a71d34994d5773a4ee664b22f783afbed0701f2894bb64433a1c9ac17658f4e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "6550a716d6dc3301f05322c72402578ba84f3a4c144ed49020158ad83d5b6356", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "821b4a9f2cda736cbf57f3dc7f618358f9a73a456de984f0785af25105fcfb38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "4bc22be450f79197669af73f4ac4dd0a8021fc182485fc20611340ec33f07158", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "131ed972b5d22f7223a18efad3a8980a98882a5d4dda17d1bd0929aab0cf3d43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "b8dd1e88a59e1ce65418b72daa49148beb7da756aada0a0bc9ed3e8c5f20fe4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "26274fad6016e0ea23d6132c4c9d62a1d9ae2d9c7cf37b149b9d3e3831b4028d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "3b1364e7b3d0618b2d298311dc5057dba75290ce2962a900b66f961c0222c440", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "28838f153cc09b47ae9a8861cb638fa4bea027e8e71823396812dc880c757017", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "89739fb457cd8816fa8bd20fca0899f3f98956c68320a64668e991ed51c7cd1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "83df9db9c8d6040b6b25ca4e5b95d8a90931e8c106cee1db5fa9e1fa01356dcd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "5b6ad9c5e26bca4467a4e45cde14482978825ae40fe05853168c6e38fbf1f956", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "98752c96561b5c6fb1914ebfe846647ee29f3d518685a63c90f36d11be00c676", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "2f6f5b701a472388bd83ccb7f174277cef0022c1728c4d8d514a15ef2ff7d706", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "6d0dc3f80d303cc2592c62e9ca729cae860b28703671b90ea74072afe1d8bcc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "0f66a6ba4a0a7efaaf69327a42aaf90e8264e24d05ddd8fb2971f406ca357a71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "ffef417a8e6bb883ec54f77935380ea9451fcb26013c4daf7a12103dfc6d1ce6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "643fbd67c8c97c4125ecc0f976f2f9b4a70994bb9dabda034d32c9730bd185db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "2e57840f95e400f062076c80d9ef004349b4c53176ef7492a97405cea949bde3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "5de07686cc2e382bc4d5c74b20bfef88454443730bca89fe293b6133927dfbee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "f65ad5fa2980dcaea258c9b8ea7fd6108ba6ce2786ef4de095ada82c445b64cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "c13e42d3921365006ac020042e5f29141285c3e9c809042fd122d3db16f98168", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "c061f34f8b057f8c09d0721da19eea0cca1b4dabcbf766dfe3e95c17467da9b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "dd3ed6861058f57ab2c5bd59424bfd2d6500c3cc332e452659ab85da213927eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "c6eddeb6cf13f33078440eaa662faf225f283c6f75e4fedccec526f17b97254b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "f55457a180c3585283328c24923173ce350afd945ed58efb3a251b07518d50af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "09b27a628090d3b587795d9823ea268e2ef6b46062ed1cf772140527c08c85f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "bc60275274ed6f032b94f6c46a5c3899c3af79b6092abb31b4169343578f35f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "e8defb8bfba735a6d07cfabba65c00c21897feeecd18ce0aa4b8a25539477e19", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "695996a210b4da805437983febca3f1b250d427cdd0cbdc7d0a602e517813a16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "a637794023c9f99e77595abaed1c359502c074593b27509d4188d3ef0b22af9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "e0d5b852e94f430ee37d0f45525bf96a100478e9d364aaa3c0048e2f31faafdf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "6b0344ea4dea2fbe91a660097ef8df560560b272faec2884b0f5a1ea9ca3f626", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "3fc29c938188e049a60b396c419c5601bdb35018b4a4789b89042cbe2d7b0e86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "2851e5dcd64a71a49c799ceb59b00293d788831ebd47abe89a87b9bc5776d9ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "f87680057e8d434791840c5de2daa725e8c5ffd0b98e06992d73ee78ca4f2ca0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "e188acd9a6e9448367c5b91e23aff94da617de42bf9a5e30a0dd60cdf064af19", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "e056acfee18cdd0669bd58239b7a5cdae9f11b683997123c9b94ffc7def77ffb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "8d41ad7448e72aad3d2d5b1038b29cd574ffab7ce95780adc1fabe801d3864ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "0841e706f216264b514c578f81f5ae34d452b912ec9396d4db23bc6026b33c87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "32494799dfefac6cdf6ca78183db5488f599b3df6d044134cc9ee702dd659de2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "f19aad1b29b33677df61ac6b50366e771fe86322e8e799c3c00644ce7c6f9d0e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "5e3f76a3c45c112e5fd9036a76c7df222f2cc6575d359f8309c4b1232078f65d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "fba2049e0791ebeef730dc32eabec2d3685786f83eb0754f4bb035d189a5587b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "8a64dffd377706d2485c9d2fe9da1e3c5bfd3943dd7ddec0cf81fa8a69cbaf5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "6c7827e99b94ec67839f749e8a415ea4e3028a6444e73474efd4ffdd8c1a1902", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "fe8380751b14427a3af85e541cc0f119464830ec004a8e5362a6436337819dcd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "8911770f3f8df88cbfbb88b51f121d0fd3c2de1a32ce31329ecab553fc61b442", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "5cee67d059ac8887ec2059044e385fe63745264b42b42f9e541bd129f7c8ff26", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "a00256c995d9f3592a7f4a0a58ed53f63e51de55409352354e01a0dc5cf017a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "dcade8f8a90bd3ca9ed12a0fc3f4901a3861be823e13fb9a856734b49a2b19f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "43d0138bb065c1252c93c2599571f24d91306c80b7595e337b47dec0835cab87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "fa06199ef73102995eddb80a4a1657d615d5aa0a70086b576945c646ac2e3698", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "01e83adc93f2d4d6acb4c7f2203a00b4c80172393b29e298db264642f641cb34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "be648a016ad5c47cff8a34263e294a5081ce91e50972f7667a41ed5987587905", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "e1ed1c7f1d66f96059414f4b8ad1c4417310105f4a5a9146e28b6c67c9be3b4a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "a44a38c311c17fd786284f1585c61da9254540ea7ad23cd55be25885c8d40534", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "89a31bd9d552643dd3f2cb7e7f462df13360e35b5cc4837e701fa7556774688d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "c1a1191816c38041dfe722d7617a1eaa7537a490dac24c8e52ac80e06024b149", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "a47127170ad96bfabd709389d6b24a6f4f34a25d7a1ca3071452ca81e26a2f25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "3390aad6d9849126f8de5736e4dceaf99854aae7eaf3b1dc23fda624bfb391a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "378461c94067e525f0357fe2452c1891b6f48aa67ed9a6f0c69e16706dc623a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "147d7aa8cad4d30f90c1c988b74a76b76b9d82b8483350ecc2801ce3517946f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "64e387c808c45f3b47ba7d9dad8fff96a1d510e91eed3a48af52f6f347391b34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "40befe33e0888ce5d878eca0d478d5c9813dd574f904cbebc87545412e732978", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "184e18c3e5c36e5602ca0c526fa8d73ab9305bd7bc7e14498ecb21473e0f1067", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "63d743f4743b863a9f51b90fcf0e35fb95643bb6eb2b1db03f7b73be49b31cab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "075cfcba2c072a9d6f8503b927b1b7a7d8da0ec70fa6c5032acfdf55c03a47c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "e40cb7ffec4194c90e1b7443eaad013ab7318572daf6c2cb735b6712f63a8c67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "67efba5b7096766ab1b82f7e58810c4fa45d51a4f86dc82916ca18bc88af1c0a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "8bebd748c62dec9357499c527b26d32fca0e7a95bd5982e2b87febfb065bbc45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "ed9b11e3bc6827d22342e686ea6ed061a533bd356aecbd7ddb517f1453ac7977", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "6b701b3e5d34b64324ba494ff284097f2c0c47ede69a66b18fbba27fe29aa1ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "843c84d4a1bff06d26f133d093d321934dd95fa03a81680facea368ecd9c1dd5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "100f972dee06a876e378b59c644ed75f4acaa23ef409f7470e7f0eea1160d522", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "35c660bce3de8227b776fdef936bc70489973e26d517a29bf3442f207c886cf1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "15072c660a43cf0c28283f03ee98dc620a370f6c4009d4287c03a5a6296c475b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "b70e0b9f38cff78b6f27c679fb574460a05b18802288cdcdc4be773edaadc767", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "8a0c9d03799ac1c27f72a24a930d406000ba077eb4adb0a978746ed8512f280d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "8a3b4bda8d2f9997e33301ae68391642fbabb6b871541ab094fba9c559b7e233", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "bc2ca90cab5d041612772923e5d2037f8ab4bd50eada384d3304c189fa29cd90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "ff6df2f2a8dff4013a63aa656b58705a60d805a8a952b279a0a4c7f9f42169c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "62dc932a563087c8e378cc5534224d18a7914c01892f8929cf099ea02a45bb74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "a7fae3ea32bd3207c0bc22baf82049bbd77d0783706ee365aa4efae53c6db0e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "04eb486901c7117ded50d1a1dab9fe9fe44f5c2e34270e087246fd5423aa12b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "9a83c7b61894e27c1a78cad1c9911ff9383c4f3b9eea7002127df660ac331555", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "76b2f89a017bb98bb55180d221cc5fc793cd58cceb5efe80f73ba664592debd4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "125a0b20ddc4a2c64a827c565d96db5b547b43eb8c05a205c32b61e4fa671dbd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "9e38fce62a61a4a3ea8fb35afe210044f60502bb503233a4123ac13c8bd44059", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "f00c92843f7b87ec451bb732dbf26b20509b6a9aaae07dccfb9a478096d47d03", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "56057c3fd7c8a2148c2917d1d355f497b21b62e5fd20613b5e765962d3d54794", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "15c836b86ffb457ef49b444e48e0d3443c1df65afd458690260c0536ce24cecf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "749825c20a8ce9a5d7b86ba7a91285bc3e63243392faed3db32e1affce2d3e29", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "f998bb2c0f634022c4db9b8c1c6c7e04f0d90385809b22d11a0786e3a1e1b612", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "3b5ff91efa16aa79894b06d72298ad9b1037313e8379acbbfa93aca963fe4ab0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "add2e22013934f3479c148b806dc23db25e8737315f4f022ab16b734958c0ce5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "74d267902e876f4f2b591a0af0edaff0e04e04c2e9700ced9b0aa63673705887", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "da75bf94c951b302236517f39e0fa829a1301d4bdf3547e5f19a1780fe989017", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "62b7918751e1e12e532078f4805215e99d792cf3c689d65922a912a8d3d6c0c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "4f8cdf33da4bbf06ad82e46ef54e5a71065fcf94527fa27f9a7151e29bd61dc1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "15b0700f7b3468d768de1a69c5dd78061544641e094a020a16d72a835fbb7f05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "b88bf3dab933fee156d4a577adc7b3ec8fc74e2aa2d51a81d390ff1b8c79dfd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "f1eb1ab1b9b6f0cb4d24574c9a71e64c34a076654ad7e183cdce49aab5f30d83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "7a60785ff7bcbb19e7489a03b891d958630d3674c4cee8eed9a967b236c24b40", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "a82a3033c13c1224bedba4c46eb33d7bde02054c9a553d33e3f1a2d46e63a736", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "3078b6d7a543250222d9ba9ed9c4f85b5fbea653fc067db45c0185299d2e7053", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "bc959fbd78d9f0541e2a5fcc84d1edcb22dc669126e4b84412792e00141a75d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "acba202cf690148994eb6ea8071aae135af85f6a7cd108537de16173699dd811", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "7bb1032ca225196919533bb14ef3269bf6470eadddefd38a7c735cf2155299ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "3646d7f4288b13211ebd1330dbf1243acb3a8b29af387d4b7c65483fe9291798", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "d49cf6bf94511b7d51268f7ff504ed9f14df3dda5861081f90f3f4da1e0508b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "30eeb1dd5d347795ef8fc3490a46739f8c2cecf51626e93e0ba2385e267b0880", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "88a9d37fb724a69d38252e7bae908f0734b9890d6f9b4c4d64d8869d2d037453", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "afc673e3d4c82e095db76aa28bca90b7f8135b93b02b65516adb4e410431b9da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "991fe17d8e573a48204a1cfcc6029df1903c13efd0459676e4a5528adbb02bb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "6248bc54481e4bda25f852eafb1b30c929ca1b832b82d9eb3410a5104cfa05d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "e012befe25a3bc89f1b01c2e75be02835deceba480f65dd1c21afcead2d9d021", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "699c839867b9f27e85bb16c19d1beec0ba91d4617506d903aef4086421504feb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "7dfac547cc8b3fa50f4155d08e42263ccf577074e3167e497d7fabf22240aa94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "4de696a28aedecab01d9d4fee42f56fd60753fba8412bfb18dc1af689da17819", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "4e3f1b4c0e0e7247ff9ba2698c154a1c81fc7f32b44c1e38142b0b486cc8b55b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "9bf738da4fb2bce739808f84a1cb88433e1b94146625c39f2546e038a13d8282", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "8ae1f19dcf3a904d66f40351a0a1a92f0a05bc739914c213b6e1d8f8e85f5c58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "d58931abc1e1a7fe730566e081033bad14ad75c12ce28f394211c40cefc96671", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "6a5cc696165131c627dd99ce77d5ddb522c1992cd9144e41b9d4809b3593ee32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "6c38909fa4a92afd385819074fa726e28ba1c57cc53e85caca35bd53aac95f2c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "97a6fd9b8663565a5a1853d1984a1d5f3960a3c7d5488990e284c2daed0b8862", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "96f5ffa068459d9a6d5320e0cc5fee317a5a93b23cb953d7a9a0dc8c745ba241", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "315736b5b46ad30b324b5954342fe9dbb0cc5fcdb1a24dcc3cc5b1767d317f6f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "a71212b1fe291c9cbb1ff8eac67544c1daa995a4d2d2043b1257a2f73014ffb6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "616250ef1ed0fde4329262c01633ed74f729c9c84f3dfbf712f29e32dc7065d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "3c95bd17b9cb721fb5c44ca56a45b8fe90b177d3d284e58f0e1136848d5d0e18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "a808f4ad1c09395b4aeb60dd1238cd514a54ca0aa4bde88a012b04c0e6b67a99", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "50d9cd06445d75c4a8e4274f913f7eb029ae020dd9192432bb24acad0d89c310", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "cd72d2271aff2f87cfe098ea0d749b41b378bee8af074bd8a6b8030dc00b708e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "80837c40eb9f80698e2a8b9a2a513c259d4b21818c2cb3be5d6e880cff7da343", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "34dc05dbc77aa644f24f748d402d5eded22f29d4ea2eebb6c99f9ff716069c9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "da5295bd68ad1edeb2a58a5a0c4eec21365dda2a70df2b2ec2b120b63180e700", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "8b72fec4d5502c8d25682d33112add4007649ea1b4556ec38fbac1be3416cecb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "4bfcc93972352299d8408e96e756b63fc76072443872976ca29b97ff7c496003", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "51108d07ee8283a02eb111414d96f61dd84cd0ce3036cdf28a39ec49eff5360f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "9b92560a11feecf4bd6a2ac0ccc5e1b9a15034d8ff4435e1049a2dc891356625", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "bdb843a39951098c57b9366d15e1e6768d372eee8c9791a4bbffb04f8afd8790", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "b53add7f886ea14354d9bae58aa1594d657c134a79a2d76c9b4f707acd96d92c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "11eff67ce7fe842b7d88fc40eed526aeb7ff18022e8cdcdda6d29655152b272d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "c8082b668ccc377755ea279c10e51a72e3b0dcde997bac49d615744149a16084", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "954ab961230f8724167c51f327392518f622965ca74c5e7af0d34ca0a6b4ed83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "2bb70fd8791eb8aca8732d01f8861b73ddbdfb39795e3549e4c92a7fb45804a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "2b1370932d6d5880be2ed03f28e0252b871eadd2228f0b0bcb019b18df0c07ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "9759c80723e4be4c7082a9ac1925b7d34eeb67e6cc11972b64643655009c1311", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "d0e134218081817a155a8c4f40b9fa72610f6e4bb6ded9afd7a6753bc537fc54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "27e2d5661a7ec8eff1f78d0b7e90a683e06f670ea3d293ecf8c02e588e67d598", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "7f9621ead6e2780f82213473fe12fe73043b7e95d65662578b0328e72b7b4acf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "89abf0a704ba9441bf9ea9117efc7d19870cd35926cf635faf681a8bea12f3b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "c6e1026c671abdd02d0104af51c742201ce9ba12f8949bbf36b4a28fb712ebc7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "25d3cc361e7c56dee284d8656a1a853157f7ae20a724e67e8c945ee5bac42ef6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "f0dc44a8033ddbbfd815002b672a15730df8915b80e77b567d99bef6effb7c3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "ba07cf55daa1dc0ff87834ecf609301efa3b2db94ed3d304156753ebd4b530ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "40d2bec39ae086636685379f822dc5912812883ae469dcb000ebeb9e32c6f1c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "b8bec1c6520b410680eb6a09515002e87ba2684c60a8ab354d56a49d6da741ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "bcbde59309cd406e826df1bf7838ccbba9d7b0b0d313cfde3b3ccae21bb159cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "319d507634b55b082656f16d89853767f7e749c4f43f0067c38fb8e873211eb1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "00ee4b71e17dda4d0155065775e6ce2f1fabf12ba0aaf967be36f8190fa843fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "1cc40cbe4ec7112c6538a5d5e06efd6a26b8f8fa6681971924b2989efcb75d11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "20f899cbadc57ab08d9da5356d2db9647550f1c65dc53228c9542a74a6d4d543", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "0b06ff6bd25fbfabb96251f003003ae4d9afed2b787244afacd56841e145c316", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 640, "447aca22374b8e88bc900ea823a8203e9be2c0245bc17717375dcbcbf000288b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "e8121bf564e82755a6d19d6344554d7c7ef4e07549f39c34d81aa96c754763db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "1ac8230a5fcae810fb7cb678514eea67df1f63ad7e66b146f412b323918294ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "72ae3ad34328a0819786e5c7fceb490b254dd901d6f516e2ccb1345de715ff44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "fbfa017a3aeaee96a9abd59b19d70aaab5b0333146e0c3196ad8b40cd4d28ef7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "c49877ec23a119bae989cb266bb638ce50bbb511d8cf91b65bfe20fccb7d8ab8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "075d8093eb4ac285ed837c40fa9d335fb4db8220c989b167f26b7de60c3ce744", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm100f", 512, "f2f9fa467ea16d162cc1097915bed4ba73c3c32b30d93bbde0a22857ff1b7a22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "61ae7d6ddf3a47c72a1dd2ee21dbcb40d0f04de0c2273c51abd7aa4d0710f31f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "feb74b4b4eee7425683c84afae2a43b1c71ee4d53e2c326f9c0e4fb1e7b9aa27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "8262f205b0749880cf5939346cd68d54684fac5e33d172a1469bca932fd43327", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "91402809019573a060737928211614f132111cd0e73ba7c80e18477857a74d47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "8fb659c8a268a85537640e307832fab1d81b780abc791cd1f49b36be512ff872", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "7220e1b6719fe5379c11c75588aa1ae546eedbf9d82bcad74cc9c007b9ff8186", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "55d35df1498eba692a6b048b1d7ca2379a0999a72aff6f918dcc04be9eb2dfb1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "151665881f42c1df850fdbf0c9ac68da55dddee4a3c7bdd4a5d869ac6b9f1dd4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "ac94825920f0b972e6172527513b059cf783849880bbfdd40c01cbbf66238ae7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "35510c1201e0afdac203f181709b41cfad38bae3d403bb1522ee4c3b29dffef7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "21bb5db792622e4b9bc3f9d7f0e335d4006cc6e95cf1b215d8503c6160e03617", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "1012855afa71c3da0715e58199fd5ca0e050e027cd7d9350e519e50f15ad9c25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "ed3818021294226b0da8a1af18c1ae4c6eaca5f9a311320af0ae6f18f5b96777", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "5fb9b0f7517957c5acc2d95af5e855652e55d5c1f20b87296cebd176317f6654", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "c3113f676ace95d6591188070e6fdd2f4c948a3ae340be8b86cbb0713a0e1ee8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "267299f7c67d9d33c2085fc1681bffff2f777277049898bee9502f2e67db1b1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "5b5ee1bb1ec90eda202009b5c703d324dfa964fdcda5806f8ab7454cc2806671", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "bdd86e9bf860d58478fe5ba7fba05b480662835a8e0e030e15ce8323502eef62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "1d6a7da67ced7f0191c0c2e1f5e6701f0973d18a31469c2cb2c9a663c425081e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "65abc051d73e039d4dee906f26e8dbd2d52a599eea6e5f8aba52c0e1689f445d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "b8f13bc500e4849d4fe629d3b106c1bc14778b418cd3126169e8eeeafe5bbd6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "5e1394dffb9c8e890dd3cf057289c599525a106a38e60df55693a7ec97a5332f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "56d5e847201e791a87ce584dfeebd66f25ed345894807135d346a953b960e12e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "279af3b5eceec1efaabad799612c900786becb54a74fc3e3e1d19983f5e35f04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "62ff56865c1acf65d398c0888e9111b4160a7da66f30c12ad1f86a2fdf5fadce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "5e9f254f367677329b1b338de8f9a3987587eb65bcc13efa7a512be49bdd4593", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "5396594be27a121b0c7987a1f991b9d88dbbabe3ceebddc63c6dccfdbceb5f53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "94e2dd87a92a6759662f7e830190331030a8c57e98b9145026b2b728b85e97a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "38067c2b44b96c2f9e3430642e4abd745f1e6ccb88964c88470d74af3c0cc611", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "4d9b8a32f4203704abbc809f779b0658d15e9038a53e1c997462c436d5fd3ddb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "53907d59d2ddac7ac3de974f6a10d9941f1b09fc049fde3132a79e3d97f9213d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "5fe594d13bd46b629ec23c7ffd7d6a950ac8dd3529c5b7a0320041fafdd03526", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "044fff6ab87fd32f2c3dd0a780d279014b562bd117ccf46cfb98c5c0b726a6d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "4b43dbe49b3fe637a5c4e2e030b2563674216a2b6c5022d5e44a20cb3bd40cca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "ca6935a6aa8786220d4e74d0d3e24554f2b9eaada3f2d723b33b1f6947ace1cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "e66351c97aebd881c0a77e6ef182c6148f02cc47686b58e7b06f69dc8bbbab59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "f10f3954e9628cb397abbb6e99b0ae5250c8a05a15b0da17317f42b064be6986", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "f8b85e927b8716df6e8130294e5e155c0a2b9dd81be0dd82cc0309308e1e0383", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "b35089d8940a7c5f39152d9ad7e7e8a3fb81710f2ccbd1b9d7c52fadaa5c5e4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "80c91ff4cf8c7ba82c974c0803cb08b0c7d19c461b20807a219c001f32b75cb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "718a80592360e31f77936859da5f0c5a2c1ee0f66b85e29189fb30796b801aa3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "baf47ab2a97cf5085c115ff15e0d2376a6223c44e3b633e3ef29076a1aec79dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "32f6d8249dde0f5f407df454d8f896a733092e10765fa27b56dde6ef8925872b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "71fb52f72f75769e46da2ecaede26b99798e4d2c6b0b98993c90a9a0f3d31a3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "0780a396858efdd2ec70eb07148046f6abac66dcc48f41b48b0009549b06e109", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "0e0aeaa85a54012c48e4f2ed0c61b179e284fda00120598243f8c7122dc5c8e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "dbffd262fa8202de458c1fbd2b8bfd8f01d9627e313e0f1dcb4270f979db9c27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "497b5665a9c4ede128cfeefae884beb6a20f96dc1cd46fde044e9264415ba444", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "0cf39fa6652b2d1375744719baf24ea74c64b829a6f02da76dc2ddabab5192f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "7c32460fbf39b8096cbabc625698584af470a70d2e3d7de3c814647b01906913", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "bc7565a34794d5ec72f022ea72eb5aa5f82f9b68a375ef76f0a549cdf593381f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "9b84985b35210e189dcb1ec6f4225af11ff9c493ec9458d73336e3f36344eb66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "bc849fed67c57176bc18e82011985d60827bed8b8de63e5b293d143092962421", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "5fa8983c9dae9277f9fafcda3c3629c72fcb5b25aad94657b2910a316ebbde14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "5bfba73eed46bb1f87b53298d79658c01e64bdb9f953e0b70ed130b9ddc66b69", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "296aaa5f108b39f4dc40029d72921e9e9b6a24167789a42ea0adbc75ed08a41f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "6a1d3ab37d1166aaf57afaaa29c8b5fead798051ab44faf85acf9fd1e70723f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "97e342e7653fc944813b83b4218c2a4d6961a8e306e6223475a3eb3b1171c3be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "3a044ce0f64c144b366a3cc66723520c6e5ab1f3e84e3798e2525b93e0691832", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "b87b1fcfff156f51bda3e21d463d9ab616902157700d2165de4b1ce5ff395d45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "13fbc11e3b278f1268807de5ed0d24ce6c17e54149ae76cfb03d2649df6975b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "38df751e1351197ca17f6c5c51c2d724789fd3034134ffc23cd85b722efb8b2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "544b7a1155a3079eb06babd0670998a43d24bb437b5335f2b38392f3d7c68547", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "635e0269345e07ac7c2f897767c0684df79109a7727199a505b63ad60036f34f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "bf1fb20ee16ee41c67e75e0c85145cda67199622da09f7a4f1f9a6747f6176c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "5ebce6b1891e307b3c08f773d94d91bc7c386d8ebe248821e77cc153c1e5495c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "9d62084baf10935f8eb11ba057e400d899def19d50b442a266fb788e7f6498c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "c3fab826f876ea499ea60db38a450e26929bc639b82e629cf01efc0d4d058968", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "3bc79764f0972bc441ac4f87153257f72f6e5dd1cd64e86f7f45b0b7b41e73e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "be325fad7da13343781ae85700db4d5e627000d8feacb856c2bbb2dec13b8270", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "9224ea419a3a10035b60f5bd2b17868f6b5b533f73607e6859f9e4285b486f14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "11a09ad19d9f4338985f4db43fb64d81c53dd30bc346d69ad6e4ab9bd11427fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "75f7813a17ead54200e61ac02e11ce2b380cbc476db57b207cc98f83c1cadb7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "0c3558b51a224a23056ef54a4c38ac40d3b1a0c1e6a975d61610960f92a7b863", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "8e49c423b6fec6128805e8982e1cf0786b9728e2b627c673990212fea47f0cb2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "23c5675b9a284d446708af603e20abd0bcf2d949d991cef5450856814413c859", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "94d371a10f62bcbb50ed9027eadaed4a5eececd1cbab3eb280849e3185645a33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "0f583c1cc9b81f8f9a698b7eeb94ceb02f6b5f33c4c0027256af6dbe8f7a5f07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "e7316832e68d8461b8ca9c009405154d84f241faeccfb7f3a6b17d3bb7cf50df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "d761abc3b2a317d4197a20715b82664fbf19f85271886530af814e8c20016f6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "f2ac24c9d4f19ba9877e5f6843fdb7e9e98a1a642801277d4f78558fa3c1e3ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "82e939098d2b2060e21cd4480aa115693001b1e8c636027efa50b1b6b873fe50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "e545ac178c9ec81a777ab388dc1cf1a9f0e4989d553a7720a2bf3b024425b1f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "63e3b24b6a5b9fb30ecf4955a079521f851484d54e0b89c4e7080afb32aea6db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "9884ebbe42fffc3e0d58288e2c7b3557fa1ba2a68e3e67cc3cae4920836a6a51", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "c6b7216dea158358d47d6a27649342a68629f65dd12e6641a892541f4913a4c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "928660d6d1114948e0172b797148f4d163f7087f75bce398727105a1c2d87fdb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "ad6772fc15a2f30e26745f067e841caffc550df36e0825141e6304422721a2b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "451482602dadc3e2723fb731c430fb2ee8cd4d95b7f2bb66b808df1c7e0e82df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "70a3ca2511c087940a9d678282438d7bcff65bec4864fe2f7f423871714d83ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "3ca008df373b3aecc9748e53ee7548051bede752ec03da19ffa6f395dd1ba7c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "dbf1d819f33d2c8a14edfa82cfc78dc63f7005b029edb76015681ec49467da7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "b273a9d8ee4d13bd1096efe38b34e3fb5a52aa282f47c085bce257e8c691091a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e92767ef7ff16436df2865ebe7554dcfc6e54869b9db3222d199eb1f04ef6dfb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "8d7d43f45265f92479e2ccbfbaee9c9cfa4af4c54aa34bdf68ffe9fd2ee940d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c342341fd96afea8637b06a3cab880ae0cbf8d3f0d675a8f30d95f7f1e9f1045", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "50deead4c51da144876724b11625a5d96cf52f8ceeccf3ad025e01cec3d44216", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f6e5ba255b6e491a2dbc16b7ce6834babc5f31c94af5f2677baf6f7f01282a57", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5e0769f70b8606d4ea8c7869af18240324e6d6b84ffc4203bf135cbe41be2aa3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "415690dae9246a82667b1ce2acb90e6ef086187b91ecc6073a2d4910cefeb062", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a51d3f88ef844e71cf408d649977e10f572026ebbac0298981801862df748e5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ccb77ad7185683d45520b3fa66eb236689aff7c5620316521df2349e494df574", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "9f87b0b6522136950963b9e626faca5c07acd25f11f7020f96ca799199af6202", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "58caab08edf67235608fe8444cc8be5f2a0010a7f1eb12e041d024105ae1a98b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "0b626cbea400ac026ce3f1967f395e23bd0b03841733a218f7da18cb4e34f606", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "f30836a498b3378b5a14d4f20ae70dd8113d6b453bde4c5542bf4e205cfcf377", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "04fbb8e7563007814675127384bbb0ce3e1e59ef3a1ade4d28f18a9a47ac5606", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "436a4655b629de03fb4b7da82b5cf6cc48d702809781398b4c4739104bfa490e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "159ee556c0d3a960a5d6dbdb6b909c34d5240fbed1ada0f03dda6e36d70325f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "abd29eb4133b2d749eae0a8ca30ee299ce1107e80f7a92c8f8896bb989af9f47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "697ff3da69e077c56b1c57447713ae9cf6337741d9922f0814ee8bb1ce4a105e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "80907376b1be759a3d324fbf03e31454e5568b07abe4fa7ad3fe0b3cadad0e9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "7826ad300309cc3f7f680472d19059c082b81dd0b3fa0fd4e42069ea5441177e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "50924744b0e2e9d88cc88888ed318ade677b771ee72f75660bb8fb950b6ee6aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ca41e695632886769e5aae526e7d05b38a26796da954b6b2be737442bcc3a25e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "423d41b1f9b90273d08703ed9a037d11b63f30b429b0c75e519bcebe232f5bab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "8872d0cd501fec875773d6e213d2a44f60cbabd7ec5858b7b5ce664a16102b66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "30bbe2fffbaab3738191f56603a4b515ba2b62448d2471394d5926f2ae8ed084", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f24280287808e294f027aa27aa9465aa5ef1c21caf923097dd099c3590d377ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "831e6185839b7790bdaa64a6c1e17c4276d92aaf6479b7678be04613b36d40df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "92c29c6c4b78ca762b7a1e07e048e6ff29bc7e912eed66f52065f7b5fe64d110", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "272b63fa776e831f4cc7e6927d9d1e88fb7c4b2d844a8ee1728655051b5df414", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "1417a4ab1a8de868713f65df95d08607d23491d389c6a481b1a451c2e9aded32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "ab7801274f2776d17d80300025cd09e77ac3a2c15672eb322bc657c478dded47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "500368f7ff305d19d4826ba588a887a2f84713bfc10fb8776422850bee2ca044", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "42024504cccd23d90753e19cdd95d434eca3e8e45ea5fe576ce109ed97e46d0b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "a166d1d12804dd6e626a189558f0c302ce4592567c768a837ad1a7ff1bf9246a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "d2c581e395e4775fd31ae388313a612152e285649fa582ef18b3aa050baf5615", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f492fa2569c5bd0aacbcdbbed8ac5c42f7fd442c10be7bd1cf0f4ecf463798b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ca58b16958e81e789494c6322c1e0120f95641f13f5e9d7cacda8ea662afe47b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "488c20af9e3013477d5205593ac21956aab7e3e65666db8a06fcbeada9e37aaa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d0d7ea33eb8793d181c48881e52787874806262644fc4c2cc4bba282f413b946", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "12bf6197fd0652f43d9243d075c47b21f8b737241a5cc9a7a26637eff237f40e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a66106593496fdc0e959a5202956aa5b2f26f9b22701cdfa5992a65250b8c961", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5aa4cdd2dd668de7e6aec0e0a1c5cc61f77919f2cf6733caeff5ea17c04add34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "7944472fba835642f3f5041f122a1000887b4dfdcd02e283dc13e2bf567db99b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0229dc7b46cb3945b29715da047b1183eaee7684a0dc240f130cf3f72f0cea68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "00cea1de3735ac52a477dddb076fb093e661fcbb97cf5a03076b0d8228fdc0de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "04ab88ea65e5692059a5d6877204d22ae282d8592879f0b4807437254e3ce0a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "2d95cf0c909cc39258de1898bcd76ee8d94d487fb38229d41b0b1c0dab1c72d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "4d470909acf8801a83e83073f0b98376101b9ed7eb415e00b63b30584ac6623f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ea5ba699f671aff7fb32f02a175e21387446b8811fbcc09322e6594adbc4fe87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "418c18e1229420895bb4878b408080f008fba0e54ba3e0aab554c5e909a54fca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "1bcccbe571319649688846caf8c1553cd722c365043cfe0e4bb1aeb97aea3b8a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "eeb21ceff8ab65b86e2ab6570bdac56703df7e3c82b0d33b63123221076a67ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "f16ec83322beff8f109d206b89fc147b944bd3f815ff5b777dcf66d7ef1fbf25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "aabef83856cc8bcf7ea52d2cf4ae69f56bf4c1077c5c0b2cffaf13adec9aad1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "d7ead3272781fe46bb11f2907086374d7283b14dbf7437487cdc7b03edfa1fb4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "94e02c89596b2936521819ef3dc37098f6a913c2e04b8466bb609373bdc9f868", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "15576a2b8fc78a0f1334e1f94917a860d5d9626e219f5d9343ab6632b87fda11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "7b76f3aa2153d3c6fe22e4d351bd2bb4d3da4eee37d5942dcf5e746ca12c9704", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e15efb28ca57a7e9d30529f99d3ff7bea6bb394600bfa79cb4bd557649397119", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "7c2102f6e7fdb014f8dd0764f97a008607a7e93a5712567e3781ecd5f6e5258d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6e823d6b92412fafb6c7de7374b9be811c5c0ba40813457373e3b88eb9d2470e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "636ab9a71885def04e66f553f0535f31487692251bec18709139582712eb8325", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6d17023ab43d731495ea4d41b11ef4f870ab738bf93b7e6303fac1894a2d5876", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "2661c3ae909ca178be2f4c14987219e6ce3c53f108a0d1d906d159e68c730cc1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "464693b4e62026b784c5b5cd1adcf09b6f54a6d5a9fee681ab24664736a96b2e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "4647d9541dacaf0e0bbf7a542f12e75ab13bc00bea6ec76cbe973394cd57631b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "c28b971971456955eba83794b5cd113f3e09825eafc394c1e70a2a91d28af54e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "e4f32151403e5be66945d66077ee561c6a138aa3eb5ff6ecb0ab2067c6e340cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "2dcb2735f225b94d41b78e9ab2c26db8016c4e2c121b99cc4b3d71be50dd6b41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "74ac43eeb00e5bbd1fbbf052c69288e2255b8f9706936b693102f8fa4a57d706", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "9ac757737de7d96e1b4abae4001ee5e72ea3a381ac490cbb0cbd7c8405555bc0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "2b1fc3e700b350ea12f8e06ee0435c4089ee1abd40c4b25b22be87a61d762ee3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "36c4a3da303781ae2c0100466c8df5479f38d832a0be41f5c929834b0ce9c0b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "4fcc07d770bd70bdce830ce6f3288beacdf8494ffa0f453dcdc88da9a812aca5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "da5702b838e2f607014b0d8ea255c9e909f37acb7f37a5108a0ec46ec0fc0d7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "3d85dc0c1ef47fbf9bbbcb933b57fc04ae8af296b8892a278b4046ffa7cdfe01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "72b366919ca0f25c0a95ffaa6da1cd7a52a92213724794ab04d0954d47d81ce8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f543609367f2770e0d5bf4ed745a5eae9cb811cb894afb3215b093e621d0a1b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "d46cf8349d6741a921879c997eae7da1781a8bfb7df9c603bca7c5f377ee868e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0fe787daac822d6d5a8efb41f857665c00e2d9921510eba745c1a44ad4a77206", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "48de3bd8b6771c40d631ba1ac0e3cdeb8e25860447328d02bc9e80fd42212f10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "04176024a239e3a2160e2bff3f481b9b6bf606c31943c5ddf25c5d535c617706", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "44bfde964c2280df7802aa46f4873543f76b7f04d9fd05fcd5a505a566d62251", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "d8e34e0d06ac1a2b5c378467b31feb6ca4f07ab54229e004535126bcfb20425a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "6910ac826c98ec25f967eb007e177bef1dc5c9f03d6f923e99e6284f4c4670a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "055a6a9737f915926233f70dfcd3fb2bacfe7ca2e05649e8a06ce2fd447869c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "f8b425a795d49d87aea5e9fe7a0e49feef80420d936531cca212335edc0e604e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "97176b83767e926bc3154040def93e925deabf328e4a4126f4cdfcedf100402b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "305b47947a35d8d94ffaa7f77a77db13a7c18716ae86172cbfa33355b2572f63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "eb711123377ff590eeebd4fd1d3b6268b212db25640cea61754e4d5d4f16f03d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "0330d1f2e83c6b88f998dbdc874e8e9d3d3b51bf82dfa4bd69e5868e05e01bcd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "aaf1579d166275f8495c50addd7375282f7fba4ec7962cd827480ebbbbff2f41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "e129d002cfa48dde60bd035b990c46546a7496c18ab1f3ef36c6d3f175101f98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "fb62ca5314aa5bf48eff923162563169c233a2efaa592c36a3ea9444d4eda2bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "d32a64319499b0f1310f73ca60c0bd00e9fc99d0938451f198415269a1fd92e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "ec0d66e0be7bdaf7b8969569d2060cfecb507d48761e3773c7787fd9ada51705", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "ccaf1c9c7f92d064ba5bd59ac7b166ed5394c71926c8810e87693b80e27be4b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "db91f7bd29fb1dd0da1a72045dfcd52d890f5244189eb91e2be4d04efcbd26b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "5d0acfcb103af6e4442fef0d5eede24d86d4b41b2e440df4f4b3b9017d1d77f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "4897e773185a2aeba25623da016461f43c4e361c6ee293ed99d775600ee7384e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "18d7aefddb87fe33a653e250a39dc214e5193e08c5984ee673f8432f356cedb9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "adb1e620efedc2c77cbb7183137cf0ded64e59bd74ea4b9ff5fc3b01aaad0756", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "8c92f0396dea5a832e47355366e8892351a9ee79838f6bb272d8cfc92f313315", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "ab1b77391b4964e4821e4a5ed5ec975e8e90d633da7242cfe6090d252996ae57", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "6eec2dc8ef431630274400a4ae2970edf77240ff6b24c2b6ad7a453865d217ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "772f75cec9f97a8a9ebdad143ec6b615b3cff253e07175b8d05a188bb23bb29f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "3761d3512c0ca90c9f648d457a90e9120d1904c3474e70d101c155faffb9c76e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "51ba6ec88f534dd30bd43410762f68e6320aa01037423795c877d9d635a3d4b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "ba60e234bbcc6a5ba4a13eaac03f651e33703a6740907bd524989136459c0518", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "fbd976efa65e99d40088a2f29532f58244bdf1b540af60af9b93fcec718d4a9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "24dff18107c31cb40332c3da29bb2271dcabbf42470b7f32003f42fbc235d0a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "c0adc2758d126f4b7aac0f35aff4bf84701f92149a7a6652b5efad98fbc4f1a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "218a0dc94f249c490f72cf699b97dab15f25c43be96565717a78ed5302bc2c3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "f57a2178ad19d46f646887e49cfc1f9fce886bbc3b514711502349e93ff18fc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "2adb0a81d3cc2d590e0fecfe7fafa43c071dfb393178a54f331d2a100118e7ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "82f5fc237215b1ff77437ae2ebf3d6397e45bbe0f8c0a8e07f21297482cd1b73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "614ed6c4abfa2074596e8073d487cfe97a1d5ed4c94bb361b7ea7802f4b3f1a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "561afddc362f946ab9c9e50a52bcec223b5a188105e30528952972122a53a312", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "bee3281e66ccce292018655ea4f010571cbdc65620ce6cb119359f03c72cb23f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "d3f04c491a10ec2c5b210ba53c4cdb9272039e5af765a6dd61150bf9f1bc6ab9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "b97ed6282864ff399deb4be8ba5cd244450b217620fea1acf8942be643f1440b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "fa3943020116ecbb90dc00be23cd521ba680a15d714dea5537caa91c82b0e5fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "ad8b6c9aac6506b1828dddd46cdd8bdb8ad4bfc8abef1fd5f64c58574f302816", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "03b7bd006bf78ae2205ad1556a67162da87cca6d54f519c66798ef1be2079ccc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "36d6412ce15334f42ac611b15162db8b675d32b2fc796ff9105c254947d6a2ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "7635d08978a9a880e253a58e88ce5dfb219474668b43c10ce135e52c91351db7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "b0753a20edeaf0ba5aa2d995e980b6dca953d8a2ec9ec991596600b909fc17d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "ab8e8fe2a0ffee2c26634d4da0a6b9fdffde27721b277a18522f091749638b14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "bc4f2daca59732fa1b260b8bda1904b375e9c1e969b860f1756f95c6402c6645", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "add748deb9114ae40700c2de9f73d45ce0879362e414972359e7013bcfbe57a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "37ac55059c1da331e9d02be1ebeee4f7ab6344b7bd72cb4619a2798968256890", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "e7b325a5c09704e33494798242f538fb78ad6050e45b19b330a0c001abd3e473", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "2a4b1a78ec557d248de2a9f5e604589cdcf8743ebab10e04990255ae8c5f1a38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "0db96e949884255d8732ce5b624b843afae33d009a6e4efa72cda9cae11f58f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "22b77eb02e827cb8cde16359762b91b39a8c6a73745b57727a38c9ddaa0d00a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "e432881a3a34cea8d8630cc8704f81579a17653e8d854441a254ffb8930fcb86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "33f3881e942ed2bdb6a530f557eb29c9dc9507b15b3c4d33f0d2d8afde1ed85b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "a96baf7d7def514ee10af468a4a55d19ce2e13c21dc37b59972583db7b96d87e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "c6297dbecacd1d1b22f9a2b047872244abc9a51b3eb6763fcf88025c94af1c6d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "8c7aa1dc8395d43a9970f7d9ff8e8355d8a5604c980b002776594e08675e8fec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "8d74c221a48c10669a27b2516897163e8fac1daa130b230b3cc5acba695cf283", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "ec73bf86d5fcd9032b71d71f2c71f6607dd11234fced0f6dee2d8f24a91ac33e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "0f0feeac32339a75c25b3e2a198ba3f161eae02633748b9a144bdb299f8f4f43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "a77f3ef2305ab0d083e5b7e0e71d47defa2f58327f64e8926e22e97aa8e6371c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "4eaf3f5c3f7892eed7b0a7d8670c071318f86bab7303ff9d1873ebcfb0268eb1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "a12b37c9abd53aedc265f8a79f46c6626b7b7d9d95d8f33c833230ac0c866421", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "780e934e86b3f0052853edcbafe238ef6feca26e4d9e6bf7744773902bd39663", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "881d01e7985ce17a601d25477c779e554d30c4a5ff2c8f10a84d5ec3265b0b97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "ea597b19dfae33e028e8aa22a9b3c163af3001e3a821ab8dd77e3cd299297190", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "be8ad59622806734b14b31b12f83609cec60c1d956e317e2bf2c4a84c1b4134f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "fcf767ec916bbccf971b0da797942a558c1d7baf2146a71b233ed19fdadd201a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "9ca5d32f40d6a7e4bb388253c2df33adad316dc07499aea0f00c078f4c960d85", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "55be88a2bba6c3cf08eb5c580df70d11f4e65b306a9757713cdfdb6ac6cbc53b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "ca2ed142d351a1654b462ab55a7551318b47f7d972f9d36a01779a64a24af4a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "c17c7c6d04b4f8d6e445e5520c72bb3cff541b5252fdb2085e8f8f4ea0f6b6f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "a7e13acca1ed7770c3ce2d1c52340b918f20bed400c3010c6321563183d5bdd5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "a0aa7bf1c8221b979ac0c8b89c7457337a0341b3f6c8ad58b60737cb9ff3c3bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "882ab7a8d20f6250b7fa3df72f7f35586a0c6616924a62aa74bee42b629931b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "c09425ce4927d9e2f5d46a0588197ee4fcc36c4261c5e31f10a8b42a9df6745d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "0d5426a5b4eb2d8fcffb08def5b7ac5cfe34cd6d79f904ade4492db5f385349d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "45fd245c428fd1a9b057672077491e4ae7c0a3873b56bed113a246e42ad8e51b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "4c52a56537311008373291e88c5e95f2faec06e2742fb5078fc0e0bb78ab57ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "fc285fc1fbc62e38b7f24a8a54801ebc536e060865e0bdb9a258c064b887bec9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "68bb8a0ea7505ee7a4dcb1c5de55791104b108961f043cff79de7c73cf8f5a1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "499926ca57dba93e4de407d2717ac1e266370d7b585acc408aad3874ef632f22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "c008d7920949c44d51590ffd2fe22c3445b086fba0c1353cb4d80edcd504c961", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "8e51a1b8bb038001b24525799170521b28b50b8c3c6d37a921e834f08817ea91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "beaed678447a2e63a0ef57d06f826403b6344b4abd5cdcba05c94ad7250d0113", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "ead21840a6d51ccf8558d57c0a97e7360020d15cf43291bcb297b5d4ce190d62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "1ce3ae60dc6fab3db42513a3e4d04ddc268b85aca64e8b06b70926643e6adc0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "e90004d64bc21f457f019c581c206e450316990aa19f8ed0e0597c05b761326c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "eba4bc064bf8756192f3cbe028a4552fb9c3dcaf2f313c540ad845020f96fc8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "9d247e924f18b8a4f945e52acd848e1a66f73a25b759525292a69a5e820851a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "f6e6f08ce9c19643713d32bef3c0527a53eba76185ff399a7db883abe04bc9a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "3b8cd19514395d3d3c89af264f09ac98be7101058ee9b381ff8e9c2447af4d88", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "dba8ddbf2dfbf254d795f2105c81586a7c6c7da500551006376f0753e878de16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "2da79637918f4b43adfa60f96d9d936ce50326e4198d604bfaba2e604ed279c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "2b5bb45cc945fe35fe3fef4354b2399adc6082c6989e3e2d734bd505bdc240b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "bfb5422f31dd70c0975b9c7e121fd11e38ca252bb834846d14dc062302b177c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "305a6b022b49e4fdd0e941df4c1f366c4a4b86baa59f4b556486e71574054101", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "b7e483a352d376a22933b5023e244ff533d43144f9a75716be8949363e6a3070", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "6c234a6a6bc737de525431eedb99479ce968242a02e4b4fe01983279ed9b7f53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "74969f910093132649b4963e367117351ec676c1f9922323f7a6919fa26c44df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "8fbb46d0a454a37d1fdb669c012fd9203c575a2976a94d547448b79fd6528572", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "d056bf9be454a46acb3abe33d6f0cf8e14daa3945c00fa48bd10371efba2a7e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "7237089feaef753d2d07dea4e745533a3959fd3da7fcca52d0e58d6e964105fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "0d803d83d2505203726b6e92bcd2a7d2865b3bf0a9f4dfd1c7dcada6f8af120d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "f9f07f26e5bfaa79dc850e06ba4eaea2c54bd0399a15e9b973301c40741c90fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "6f66b1362b0ff2b534855fc6c6bfb73d1c9e660c1dfcb4a820d25f72f6ddee93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "44aa6c00f436813b5504986a20282aed4556d6bb3f33ced8b898029a5e824567", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "e1c2b9ec330f1f7a2f05aababeeaa5622a8f5726f1ebd31cc070ffec5879974c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "831e3bc9e36b6f15611686c9400eaa343ca8b99e143092a0d46c41c8535d2466", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "b37d85ce6c36cc7445c6e446382280304b23f2d28b8fc6ff12ed9fd4c8c8ad30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "34e14ca8cde26daaf76ffd4cd7026eabed367482fb99d0a9ab2fd0b07983d52c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "3d034000fb9f565661caa33f6c0901508b77d449d3267fececa7846fb4413523", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "83cc802ea28e3e13950021af48a45af460b594e02d6c23753c96dbfffad98256", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "0fd2137ae083e2bb047e135e2c4b32f55c14dfc195753e24891356cede128c07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "3c20a670ad682ff46a967b05e7e475455bd23b97085fd193315998bb100726cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "cc5f50239e24e0a27ac989c44277eb0bd3d5fb3427db0f40445e859ccc9890da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "675f01e636f0bdb20b59115318aa33ea21bdebeba8b8d704d364542a7bc3fff5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "518db64da18757fe480f4d1939abafd9f8c1e7b47fc78d500eba4cd9f0bf78c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "caa32902116b927da3e24d706e5a0dfa7affdd444499ca16e8cc424109eecbbd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "e7fad1f23c0701caa1d60b0aba7c46be1f24f24f037aa08ac686508d1a188efb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "ea362ecff42f9b45bca1b6856476ac8093dc745af4130d80ac8bb820e9dfcfcb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "4a71b224d73a3137871e791f1feac60c908de0112884b58005312b69ae107208", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "99cf8d2069e59698ab2cb33847d081c5f2cfcee53532971e44b713627c531f69", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "b9df7e782d4f408232d0087b26c853137ac2d10739dabd6007f84ca0072f9505", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "8e98bba028931a9b6d91c47376fe43e6529dddb5073548f913e90099023fae6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "1d3e743f6aee9b79a062eb0519e1aee25df827d40ce34c663201e8dc23e0e78d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "a986e7bf1ba12dcf972a955cb1de1a7fc246b15ba7a4399e780af49cae90ae4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "f911c12532359729ed529d76d5a02990f53ca5fa7a698046e73a7a7e813cc4f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "ca37915ce49bf9ed8af9e9e7824813073f6ba64ec4c827fa5b55864847b77008", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "3b6102e9f43dfccd5f1377ea511ab86dd4454425edab91bfad4ba0070766390b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "10298af5e0dd4c666dbac87a809b66eca03e997bd8758dc4dd06d76a2537c61d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "88509ca1f6a9700d7587fe093a19cab0369edc34c485cc07c47f3427395ed9a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "fe66c524daa1010cccd479500d6768c60b2386a79463a0c6616f999917e5a7a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "83d663039e71a1a8bc0563fb5bdc3c05dce79846d0a5a19585796e09c78542ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "f23beda3e661a78ce8e2c32b660ae65c199ab4090f2efcf77d1b5af13e110afe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "f64e4319feed1a88c22326d9004210a0508aecae25503591b41088cfad2c8eaa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "9eb4eb1595dd0fdbfa9243ddc1b436bfb60f028814758d59b17ec7fefb145626", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "53918369353c36d8c6a5b69e0b2d9081bbd30d7bb0ff13abc5bff45ca67bf064", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "05b482cdbfe566345c70d5d21e90310986af3650bbd4fd842ce892ab68313d55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "e99a2c32b84058766d6b91ad666678dccc8ce4993d973289391b90de9e436af5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "fb3b3d1e4085aaed4922054589424e263d796d7ba4bc9b934094c84e6248c41c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "bcbbf9c424fcaff2c790b3d99e0b758941be6f857b9efd61e62a18da4e478e98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "18ed31da28645b70b7ee4399a349f9537e631f4e55defbc75852072bf0fa785f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "b9b134bca351fcb061034a63cea09b161bc4b4ec118224c1ba8542e0201e9006", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "c5917d78da68b8c1c153eda7dbdd5e8d564f2770d214cf5d7b6def331cf4445f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "08315c80f488e6b9acf3ce15b5eddd39f530a34f1a1dc58f202da9c0e6118e3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "fd79e838f259d478a6be20c1141804fe40309ae5d2660b781373bc3e2bdee5e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "32e52101e36dee4faeebf849234dcd84138085d2f204976018e17d736f2c6c19", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "49c2de8f2bbb7510ad51e96b2b7298865d399a7a163c8d30579da18496623978", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "83b2e22ad056ad36ce11d5d278b21532ef46d3c1f4e51b0bc66eeee1e9e602db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "e1b0994e5d6e2b840c2b3a992d92912bbe3919d2ccd7855dc451159a70c22225", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "6ad51646f0e90c56a4d42e3f99bdfa12ae699d98873bb5fd8c518e2d1b595250", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "ec281d45d76e95c8cbe0a6c66311f165963344fdbbe56300e7203dccc010a865", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "042d4a17a9a915dd870e20a5bbc673466ed9f40d57618200cbb3f0c981caa693", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "647bba09bd3c865829112601464ae5657104a3f5c9103f3fa6a0775fe7e13ac4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "b86a61e16b08b4793bc1e2da078a29e82711cf62173a05e5eec0092bb2b68bbc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "ff376d198f6a26908392adc4e24c5fcb20fbe7b4c8469459bc1c63c1efdb79db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "5b441ba8f84a9adfde623174fc565d8291121ff30524cea6079a5870e4b769b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "7ff2c2e72d08b85fc1495e029e925de83014973c26eb14ca730dfc3d01b0541e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "efa9625b642336d874ee636f4f0994dc99afedcb741ec4eb413842ad08b45d34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "894a24d88bff0c8c1d4e459d75a86e3a3a6a80767051e1e5d3a6a18cea7aa2ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "1d8de65c5b044f9c40c122c89e714f1f7b9dad866724edf453d6634a52dc7c14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "90a5650b7050b54a5b856c7e6b5e8cf6a8266c6fd8a0524da46eb786c672a07c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "281c90b24dfd233828f5c67cd1aa80d52cb7c5c308600f8e31f678e226a1104c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "e8ed69fd3b1f74baf2463381bb75f9a3deee12140b9e87b36e0be94fd0949f4f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "9d18bf1e8149ad4a15a1501e704f98fb13e4492ba0d5a19649df7d6b882c7320", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "3db3f7fb2989f0dea6921a4275e81e16a11e212371a4d0ae51f9c7ce9c50132a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "e499b69b700294b9d3a9cc17ea1c204a37bcc24b5fc442135addf0824c35b498", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6b263da42d0b36d0e4573721f6a0b185e22a0639c88aa3b6f62dd11e7dce433b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ad20b98321884578ef2e7a95ab7471362b705aa46921da43e2df74b387c6bf5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f53806e386823b5d92322e35fdf59d0d65dbe210df2f0f509706de45b40305ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "484b68fcc55d5369f5f2d1afb09262c8db871cc8d310132ec8f1af97f1e67a74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "394212cee4ea02bb8c8ef2dd82ff676135b85762d3d949734fae690494e6af4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a7a5c37d44d7efb22cf379df1ea250d3b4317f87500c4bc4250f558b1ed31196", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0b5b762dc58e0732614cf91d976057d3a72802ce6093d44962c81f044a3f5acf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0dc70f56cafaae65703537f6d43f0f78b1eec3ab11b94b61a76084e844a3876b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d26925e772dc504582061a20f8b21417f477a19cb90fd01082a1028d37d4524a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "840c23219f39e7644c2d35dcdf60cfd39babf27dd85502983c0db53fef39529b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "4c122ad2df320ece5d3f11deb163aecac43eb444ff8cc1f7d84c537a530ef44d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "7e7ac756695179ae43dac8b9e135888306a9d09a284463282db05807c990ab77", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "1a0a66b9d1345e90baefd8006f061d0d06e98dd05f7328705304a98d32009289", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "e15e6d426bbdae7eaabaffe9cc9707725a04b3f4260b036a6822004f4138871c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "5e744fcb0e2fed32d207dcc6034f8407c9a214cd02b6d6c6e2bcb3bd2cb9deb1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "197e6af78e82dca74e3997ce4ea0a63e7708bb43841ddb2a0ef2310a5f35044d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "4f67e346d782a9019181baa6aaadde5f7a318b0ba0f5ee054774e614c74bbe18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "f3d15d54a694f254360fb7bbc061d8a00811ba970bf650871062129bc8e65465", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "905796562256082eff975ec2b56a9ff96272092b70a49dac61020ad8b4f4aa86", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "114c9fa392b4ce78f542869c3e6b526f8829bb030ffd093beac57789b4cc9dba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "cf88fe76e9cd6b4d6c489fbe9d67901fecc57c9237dbe9ababcc57177901ae99", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "1ec4c2dbcfb2568c9191466717877b03667e5f80e2c4e0d84ebad81664fefefd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "7dba38af3eb9be245b6c72a21f055f641d46e6fc5479f6502f43440e44dfbae4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a8f1b71061ab7c1b4c33f5fd065ab43c233c1181380cb73cb7f07508a0bf5291", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "cb31bb8eb87e5fa971e5b49f528e72df25f6cf79474778d26a88f3346458a3b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "d0ab804ade2c6dd3677986e6e9d63e42118dcb0ca4afb1927f68ad26f2460436", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "87f525c3a0c5bca64a65d7a08b90cd9bc1fcf56d4c6398c5da9a11f9cbdefcd5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "e83183616e0ce4329f204409164144b9cde294802eb17b66f9638431bb74ace2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "40ce63c8731b2dcff83a1baf72d2843172f242cbdc3bf596b2b7707ecd7b9441", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "5197fbc6b998832c5d13b5f3ab27bb2a52a576c9144392f0440ce1933765cc5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "678844faa13e8dc7339394183c2f8ec68a1bebb3b15b304dc45c3252244fa527", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "b1f888e299d2f18df75f2b1a567d4ea6a5a69d41777beb34216ed4098557bb2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "65b3dc207ef4486b3e4fd4ceb3a03de5acb7f0bc3580ca7ab56cf34b36968837", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "dfe98d19b2930ee5bbd3758af055be086ba9c8e19db91ef0440cc6cc447b8c37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e81afde958cfaa9bae77594dcfb53f94010c0bd21307217aafe55b23f6ec04a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "f1aae208590094bee230937d47915ebf09334a0fc7878211a683cca617fd9e5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "1d64e0148ca2fa56b835c8389528bd03de6cd195dd7759cf87f4a8d9c7a1c2f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "8e8b16d9c5899b34fd2249b8fb668e52ee034048ddd8fb9706c67534c6ad0103", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0a7425389e67b23c3baa99ec11cee774e62f4c767080b785f0818462375b8cdc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "af3699fd27632fc412cc7dfa5f86a3d8b1151f701b08e39870bfca4311b0f2a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f9b3696c6ffa4d874a9f133c661cc11e948f75f1c4d63085662035e774b539cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "813156e6cc02d3034ea41916ca1ad8cfb60e072334d829cbb47617764b4ceab3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0a8b8100ce51240be552bd90eb84a2f3e3f235366e0fb840ecb336f9b10b7c9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "7b39d86594e818954e51508518e5440e6dec5f28ab3f0eb48edc4fdda8c91254", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e8c2ef072931ab7f5b913045ee8cb6f09c50201fb160d1640f3ae21e3eba3085", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "12e6108168b7a77ded01502e57c23897fe55c5c53c6db84602742767315a76b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "064f105d4a69f3514c1fbbd052d87d3a64539fd8a3411a62036310d26088946b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "4205229dacc45d22065c769b6d23b3de663bad96e60bca4d9abb6e3454ba5811", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d66452a49ea348df0ffdc1a243183b3305eaa8036b17c191e711432b1d0abc73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a40318486ceccfc33f543ee6f2951ccc7df90ec905ad69a5934746737c135f16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e3c0f28b00e6e2fa4b24ff3b34a8bb616ae07ba49c0648a0a3a742c37bbb9322", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "63af01922eb3f2273440f775fda787a211cc32f07d0f8a1f71a4031deb27da83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "01b25583725fbf134b766fc07b2c59ec6e3c78d8f3cbaabbaea3f0de0b2fc4e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "43155dab5a9d0f3e025b1a738f7df91ce96c2b429c00d8653b1508b9083817de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "4dbc4bfadb39fc2bb22dd9f10b5d6f726a2db9a2722f3c2d990d51faada7abdd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "892459589b1a9548ba8dd4efef75d0cbfff1b75cdf5f08b9beb14eda1d1b2cab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "7339f730eb4f0dbace6e84b7b1976afff6d386ac2d88c3147f2cc42c6777ee72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "9a1cd1d7ea2a95facd4468e2e7b037b274545323cd0fb317781ca2525cdeb632", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "7bf6b1ecfd533f9d1a29a353e6fec65eecc55ece6917a2f49b8740ea79063d89", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "9b80b2ebc5cc56ad66086a60191b6ceeb22642de3e933debe38715a7154bb16b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a146fbabc22fd4319ddbbfbcd687a8f600e1eed43152f6a3fa74ea664d4fd025", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ede2b09ee43526a9d20dc4a96de8b8ef1b4e77db224f331ddaa56223663c67b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c4d4cdf55401735e5e246598d153999cce0e7b0f4b0464c29cac6de7287e3a93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "b9fdfc5ae11c48b15c491b331cf4c91e154c84851a963b2ec6ecb4e152ed5ed1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0e9ac79ceba4fbbbf0d67fe77b3d6715ea55fc294893aaf0de8394702343a725", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "532ecfdede27899fdec0a2f33f9954e5bfaef3a1bd93bc3b2b971c09f6935842", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "8c33009086f5886136305969847300a68f1e8503ce1b1fdf75decdcba2cb6fe7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "8e1928a8ca534ca16f45948e1cfd3f7400f9828e0a34ff380481072b843eb43f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "5dfb559254edcd34b459d8400256614c7d9943454344885aa0b1665fe7b42118", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "4dbe948cf17faa33407d15f285a2ec65a751cce4643fa9c3b624241a3080fdc5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "29fb6ab2547fe4b99d8190b75f67df22f595a3a752ff7005727d477745b8eb7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "1249a4cae5d1673662d9c6da72b98dc943555e18dc0752604592cc286dbe9ce2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "d33efcb00b88380e55c6ec875b820715de307a265a9fcc6a8726fc54fe1ac672", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "a2b8a9ebf6d7aa1d5fadaa104f7266356d56fcf1ac817a5e275e9a6d87343942", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "f987e188430876bde2f8b5a670f80cdf46d0ddcb84efb51c3b03f6375ad5637a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "542bf5eb06a26f0fc2aad59d3ad6fb99b744af9e99205a90cf4822595f96d34f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "b74c4e40a92d50e27c56862403cea295f19001948f1669e0d80e01bd5564084b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d190c78d7acbb33bb670ae7bccd10ba0130bee00f2119e5a9a00fb0f10e790b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 640, "fd7e4bebaf53a923b2200b1a902db59d1abfddeac85e41082499d3eeb5104c56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6ce0e8a86d353526a845e2b1542e3cd51e5877b20e2fd0e93c3e56a09778c73a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d1c7e517ba9da348b6e79bbb1caa0668f547f1144111616b46e5dc10aeabf5fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "f60b213c19f66c3cc232b56367bbcd717faf851ff486bea8e903351bf148ebff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "eff22ceb054ca5843fc4f2093206dcea44ac504b0ca902d1600d0e95420248d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "8164244524f952967cd652ff7018578d4eba8a8b5c54e62fad01b8f0b7082795", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "fd0799e5cd9937deb7df7d464e93663824682ac5f46c39b4a5a86e5991bca8b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "84d915e06816cb9770e87a3e79d6487df99cae2574433ea293709d476eb1545f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "c4fafd29eb90f6b6178282aa03cdccf82692344c46a5d74ad4660bf13557da39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "6f11a298e4b8486d3009cc8da2f1c3b1fbe4fbe744752c0b05b13b142a8c7aba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "ae279bbbb044d98f3441c87b76fd3270f42b99b705774725912610c74399dd1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "62817fd1a08c20fe7bc0854ee6bba670979379ce38d8c8ef92e40914ed8de4eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "a02609e1eab93e43333a259e94e901edcf456fc08709144c58fd44db3f52cd34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "b929ca8316bdd647bf3e72965161b06eb02196f055fff7eaafb5183095cf912c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "1a654ede72402ca7ec388c787d522ede5abfd14f7c2d74b7cfa15274cf726e9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "4fd1af3f0b2feed264569454b495990eb42c8c2d5274b106ed5e13a8bee651ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "567271e77ffc62eae2e649f2d95a6af795c90e28493bb48605a07574d3336976", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "72c426fed68b0542db18185ceefdc076a554e93f9a5bf30d4b255d0b81611ab7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "adcaf666d7e1d705319dade600e1b7762b694f129e01f9e3a168db84064074da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "799989136257d34b458c16352fad1cdf4988416739abc486085730c02fe9fe43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "5c5a07935d0bd73f0d4c5e2191b6aacb9442ab1cad040050a703281b57a39c48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "acd9c0b4c29c1c17bf80c48445d6e4809cafa1a60ee33f8145aa600ef485be16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "183d3bec241ad23c63a62b3b8917be8e698b10dc026d04d25d237bb6e85060bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "10592217e83f63831e4ed00194a1c38d48534c7f890d5ff7861532b2e066cd9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "88cb1bc30b86b024263c88e953307028f71d018658e02f6f2497625c6538122d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "92a47395b759fa54fd1a0caf6299f90b642488d0d783c3cbe1da09d310c5efba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "7d6a17c699af7f985745687ecceffdb8358c4a427f0cb9a8d7300ea2ec9f6eee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "b79cb856678f16d9d061085e5799e7670e6a4049a0d107182bcc8bcd6b2e7648", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "dd54be4d244e186f09c4134251518d68e667b9eb9146d9c0e6bf0676acc03c80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "4676c1387d1288496e3931de044090dad8db06f2ea448d696453618b9320172b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "2e1a791c8fa20614ddd8af987d2c2ffe1d0671a191387eacb955639b6658c04d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "954165c26aa6f29423f8a994614046e649edd04c14e57ac008c650ab91cc0be0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "e108ff79f252a91bcc29fc0b01991e48d50e0814e57536a6e4d081309698c806", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "c2d96648f5b79591cb583d6a77ae2ff2821238d0ea8db6620a5754968e1a9051", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "9378fba8050f3508db556ce5d5b0e3865fdc874f071640c168a24866a4c3d89a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "30dd8c4536d66673149ac6b3b5bff55c799e6862f503c01dc15abc3498c90055", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "e4bb4ec1c32818e831252425071f38e62a767f3b5eb3ca97fa6fb4cbbea99897", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "9b7c9bc9905cb9e0de1f75dff257b3e5cc402984860a30e137c2883d8c7f2d01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "f4fb0ff21cc03260feece9f55f278e76c3e1d4e43c6a3379d19e231fae930a44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "0d6e3a3dffdfc6d0d65b0ba51688fa50ec4e490c962bd58d2566e26e4668cc7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "ee98e56dccfc61f65d1b7c9e399cc57a6020439a57225893bcbfac67975dea61", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "f7369e805239bb8343c953a6f0947494d9cb4ac898e69ede6f7ea91044291697", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "379a377f4f2997fe839676b7f9c9448aed9efd4650c947bc2a9671046214f951", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "dbea106722b607b550d8d45e1fb1ef869511bfbca88e680aa5f95973f6b839a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "9ec2fc507f24b3166227270610845b80f0409da8e4970d1f54959cb875d7aa9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "d8da0050b3385fe69f1012e00733aba7f877182b1275b41497c34240a86e25b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "ad62c39aa3c33a330b18472172e43a7b2b0f910a78600407d314674e2ced6253", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "d5c345000f5b30650df999d1ac36a51d3508f904cf12c7a3a74894fe0bd08f51", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "5bf7b5174607d9e0b21572c2fb96713e50a3ea78ad9c70db1dbe7d2f676833c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "c1cf0d491f9bed9a8d1010581269e3e32a9f4fdf53b528bd3798fdeb93d79943", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "3c54c919373a3dfebd8e0d4c5814a8e782599b8d68ee76f7ef86ab4bd68e505b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "baea224b41c72b38de1102ec01a85ff59f448dac88cf66884e5dc1ee95ec92d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "65c333aa43151008e58ab798605866bb9237b234f25a4c3aeb6fdafadc217bb9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "4aa2a1c54bc95a2698cb9dd0f33a28fb0d64ef0d33fd3753d7a60808ad49cc9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "ddf524e7d591c709bb96135939cc3ec989c482b2c965219cb580b12d45e712d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "b71269c7bea60b332413fb8e6dea1045b9f73a21f18a225b0de69fa5a1d5c9eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "0b0807e55d703956aa3d8d25d8d7cee774b620441f2227d02c0b697e55e466b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "d75745eb2b1ff1f1f122a47a601254f2f546015a88c5fb44bb21f77e6ab6b8e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "8269147bf83711bb17a5a0f447d1a14445411776915622def5b1ef87bae51bd4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "e89c133b447f1d1b13d9c77ed7c12f9e42bd15d2589f273cd4ac39a6bd4dfa30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "2f8c4fb5e57dbfd6c951aa13fca89c120d4735397ce8c7696483c820480dc9b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "83bfe9a05695cd96e802d7abb868622945d46e634cbab07033da71c2c36cc257", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "747eb0633ada275dd7cace0060439c69fcf16ea709658871061ddba0acb5ec75", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "82bc241c20d9180ae7babae35406793a63cb5ed2f77a69dec60ea3590167139d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "d215a3bae87d9b8a306f4faf62093837867b52da57cef764800579aa8b03f858", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "b07b22fac17f814905b18f425f97e3ed7237e5824fa743d43eee296e2e3c47fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "73cb15c4691cf75e6b5d7a3780307ea031fadd59ad30963ec9a05e9823dc9e9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "244ac77d6de73df8f990c1982606a6a7d712a0c3d417f2de9dec770ad30fb183", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "a9bd044945c55f0b20644c99072d2fa409f78733bf02a615095a2a53a6a41dc4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "cd03422c64ecabd71c765b20ca5a1079d03f99fb459c00ec747ca9efb53b835a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "b0de066f21861088dae54404f7b2bb5e0128e16e38f456c625d07ad645ed48f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "361e6151a10e7be35e36fd44ff3a9d5c1b5095c282ccd918a31754b2121fc19e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "cee53427e7e0deb8d582d192f275b08fc8e1ce4fa5ccfc808db4d2b16fa92207", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "7be9398d6690b3dabe293204ad11df7fc9c8bfe3994d36e2ab9fe8c213d77cb6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "614a3e8526f25a1614b5aff271dee6ea37beee30c1cb26c89741d76924fe448e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "ccb54add57b793f4db8d1302b926d15f1f813fa03656bc78a592ec1c399341fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "5d21313f68b199e612805ddfd5905e166f1f985a4d8c2f9c5377db30dad4512b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "b21e1553b4bb6cd26674d701a83889f190c1a4b4b8ebd2c1513bc50a473eabfa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "528bbc980d6ad050c537f850f79ff6bd023a6a86963c08fd6729e558b0c17566", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "f383b0abeb642df66d5d1ead866d21722967fac2c897c20099e463752ff592d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "eadfed8c033254c7aa7c37e3353829671198a61518202ce909635283db4d2b5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "741ed58a25d9e297df4ebcdb670cba28a0f9b06e5b48bdb01b7b35feab7fe4e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "04229d9bab8dc2807a0f973d7d3937c902f14c838b41c188288302cd5bb65d1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 640, "6aa50eae6035586b7a44b16b8fdf3d7966a65b3e2f2752c05c33119af4666595", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "9b84e05709f08d97cbf99b14d3ba4b74feb732edf6ae53fdd134066483b60f3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "ca1da07e7ba2095ca79ea9efb0fa4bc3e3d56e802e8616b3c1e9ba46b2575c2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "a4a76143df760aa5a47cda2524fb219ab8bed73000d52f0ffc633d885281cac2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "8733ea1bec6443ad9571280db646e571eb92acb56c7dd5635fbe2f964d5207a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "675ce8dc3a9540ab8c8f26b24e3a53dfc9f8eeeade69c1a401742cc5461f1bfb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm100f", 512, "e990a31c7b6e092f54f3be11fe4ecfc6d164e9956d20b10456057437f22a7739", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "6ae090175127c338b329d41f6b9e5927e51c05f19a4a79d39b7fd651bba09193", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "4005eee13278a7ad75e50d67ef792e94904d736787ec461266ea5e415de6bc7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "3168bff3de2114584c1660cd229dccf9e68ab5db886d19519589e59e20942ae5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "6fdb07f8a27f84d5fb34b2678e9484faebb73b5fb5f038bd7f0323662ddce5a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "8ef6381ea8629b2f1a50d657b72aceeb3fd9b0e8b556018ea99c9b8598526152", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "db9c2fe6f3c6d6e96738893aa2055cc823c516c9f7adb4f5e4cef744ea0cf408", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "f0945dbe668f59649d2abd8f6bae4a102257504e244eda23e03c79a8485da609", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "00a84befd165e20226b0f95608a4050130bab2e2a4e0ffa3ee0e34d23813c972", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "fa15c045558174b63f5f142ed85e7104907233a6d4b670de701bc92a4b89591a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "3532b2ff825bf2541d5af73b0fb218358137c90428b1e8d710caad2fa0f7ac01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "59b0c4ae68d386842c6853866124c3cbda413f8e12f423c4c77c7580eaba3d5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "088004718441e4f09f5ddef5ec6eec3597800c6fb8fe3129374f6a851f34fa6e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "26f9bcac234f60af2c57e08aabcf3be591085b947e161e9e851266d90842cd42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "77a601e878dfbfdeff15d50d28bf60f49897b5ea0f6284645999e6fbd5665405", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "a000a34ccb14ee3af82bbcb970b3472bbdb98ca62e470d87bea91bef18e7963b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "37a5bd169fd3ae8967afbaa83d2dabbe22582a1b2882e0c2cd63b14bf014b6b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "4d4560ff4db615290543f130d4bc9e14b11c13e94a71a1cb2ebca5cda21fa908", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "8240f19a5429336c16d8ae242ef672aa1767c8856bf0e628aa64095e8d2ed619", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "fff6e8423d1ecd27d39385df276baefb2980f4e11d24da7528a7b2144a38286a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "398bd87fa59c5b90fea9368a67af2d09c83b060f3b4ff9c162c9df343a2a1cf8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "dff31a1aeee7492af7b416b3b1fe97a4febee9019f509f59462e8b2fea99c989", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "e14d294dba860e176e84fa032081d3956986441a6bb4208d557ab934d4f3701e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "626bd903f7c37734f498a2f6138250d81bb39eb1f48fb65f64067652f7d7b4dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "bb55c09f5caa9a922e0dfa8cacfef8edfa7830bcdcd9a8aa5f1ba01e8231782f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "7b6ac59398ba255060a6b4f93f538c1b2e886d5f10c28725813b7209885cc504", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "71889d92e900e8b3afaeecddfa1053ebd4461e5306be744bb5d458c620ba1409", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "5c6d93195b801b6e64a4091ade3401d8767222cb21c2e8913ac47ec9de0af060", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "19e92ae2d90a1141c292e6222abb39f8da5c79ce9490290c8fad60cc4ce174c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "494f1e61a4570ffbcede8a27bf059beee4aa391426041dfb1422c841dce8ac72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "a84dabfa1446237bebb3d6f20a5aeb1ab4937fc2bf1011be92962377ce080843", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "0d29dad4a8db1a17663b8e47a37704af30e8ce4eea142f9dfceccab3b4af7fcc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "e42ae039ee3af2d3e21fea49d94b13e73bb321589b4b7d72d6fe501e37a35a22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "e2817e988e863db46b3747365fd95550180e9670b8b68015404fc986907fa1f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "54129642690f48097e745ae1a74c60cc7a41c6202b1e752475a51820b649db82", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "21413190661fe3cd88f290d85e3c2dfa5a44caf9ad8b9895a7f8c82e7b74a4b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "8b3237ea34eba6e54dd4534b64129928371ecf6a65d3d06e39387c0da13b71a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "e3942ae1667b4f45d8a80cdfebb0065cef830a26c1dfab8d78f8e554df3ec381", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "308a62944cad79bb9b65121f259114e9bd774f5052d513d3de9853bd76f8de4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "73cbf5aa373294c648f466eaa4b92fadfa7375d358a9d43b04dcfefb478da094", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "ff0ddbf020f2313ab0b19d2817befdb881b0e146de496da84b3329852264bde7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "160ebd5064a99e0365f406c191dbbbbbe2fc1019e94008cd01687ba2bfb155ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "f67e7254805be7072489d953bda686f40971ad4777c928ed21b8584a15ff330e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "b21adeb6a672a3f31bf0ca529b50acfede2275fa173ace63a861191510bb6220", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "d6ab1217b3597bbee89ba4405ad06c914c8e6803b058dec4656740d774a893d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "5797d6baca5b9695bc5cdb8f7243fd5c3ab14b58ad4a36f629cc6ec3b3be2d8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "23ad285392b016017777ff1316b8892a2229d33b71667d70bf80b0a9787a72eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "97a0bcbf032da70c5ad47d091f436828b0c58a9475455af0522feb1183cf0b5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "a128c687529cca11640745cfcbd75f8c90568e3b7864ef268c033d8f993dd3ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "33fee069ec8d534ec3ee8c2567a79785f1b82b00989e3b379ddaa1c383b10ff0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "bddd0659660d98b9028607cff09c79fd4fc2982073d3f412bd24af7d2df8c0ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "4acbedba19a4cae59c8d5ef4ef316f6f4ea25a6cdd1f7e5939132c452c8b6ea3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "2e05052e77933207ecade4dbb9d006dd8c2a3a296b07d622f305a270be8b4536", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "1afea07c74f6a6a5ec4b34cce1562f85efa55ff925c199a49a97b5ed939ac145", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "b2aa9be13f6161ea19d6ba9bb8206e463859ee2dcef98e885e012762a87bf3a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "ccc0f1557b85a90843c1e52cebf94e33622c778ee49774efdb8e7499d69e220b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "a9157e55514b96159e74f2db2c6945ab446584618923de8d0d5ab602cbac2c15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "a1b9da000b4bfeca353f56f5ed8bb9366c5f5f0150750d048cfb9b1f8da2d73c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "90285e3f5237dd600ebdf17857236e4eb4591e1b77fe7a43706f73736a92b92f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "1049a3cbf00cd3cbd39986cfcecc7e8ca6c10d922b964eb9f8ea8fb009922333", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "50edfaf766a62bc13b7f784876a2980a9bf159984eaf3f2bedc864abfb542458", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "b3d056e5ba538af447fa5cc1a572aed1614a6bc1a10a2b013b690c16c51d5c7b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "56d1c66c352adbfb3b6cd8b04dfa268cb51491df7fbf4f78719d2fe23498bf62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "b315dd8e1cb160d82034801d6e03be6cc0fd5eeb904f2f89e6c340d74aae38a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "b32e2095ba09059f9cd3d362f3d21fe7f41f6d739d8659217b618c6d478cf550", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "61330ecf24ce842cfa4a0010427f536140ecd0cb66853ca048741866e2dd67cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "29c8627eb3e814cbd8f560a6cffd53a687ca01c24b7435e690303fac2a3a2aad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "8a1de25273eb5f878bf179af7290e29c75ead065080c5a08d2f765990de0e767", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "7cbd29cf01e9c2cc0949b2bd39306f752afe25d8a583dafea56b861913d27200", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "87b168539c28c8feb42028399cf088858fe9194aff5b04c39e59bbe673e17745", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "bf35865610ec96ef9e9839f891893087f7e74d28324292b841c2c66178db383e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "7597d1db9928b94851bb9bb42a251f74e3effe837f472e0e38b45fc89bcf8b74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "94dbc0cc818aae6ee12147788a588c65a9d2739795897362110db6c763778685", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "0128d41ad5e18ac5fbaed142b12750705d8b4ae289ccca000eb42738b4fb8009", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "f448a853853ccf99c799f254a46e30f27b75daa6179b2cdf6f892aba687e94c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "345777cf18c96dee6807baae4c8b8db789b662784e7cb7423b1d5002b0e4f6c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "630a62fbe69371b02623959a1a71a0b7b26524d204bd71d888bca5fd06b2dad9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "a8372b28583a2e09fbf5829978be3f7d8aaa5c5de1dcc9e9c9977b6b904fa201", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "86ddfb3685da527780692d52b6021b3dc099ca15690be9800a6fb8d53a7f5f2c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "064199d327ec14ad069e62b1d6f8f85e37adafa6c4c275e534861779f18cff81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "3597c3c5330486336d2ec29c62db1f990d858148774da7d3cb98aa4f071d637e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 640, "3a22831c2ca73217e947e24af7403920165f352161660fa46c44c430758f92ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "7885a65cb68245f0038f769260e565f7bc864509e1556f9d832890cf3ae6c880", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm100f", 512, "18025d769127e64f4f3ed18810ed2fd6759fb4ffe01fb124e07a7c9512ff3cb9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "86961ee80912c3b3f5a7c9e7a14d9bb236739786ea8e9e718b6ca91bccfe1844", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "0b637fcc1213665d154d88cf8f470f864f9f151b2fb71b6878e01450e1a9e7cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "737640ab4a18b974268017b4dd5f2fd443530afb5593bb226f0e79b9cb77b810", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "a8cd8de07f7cc25849b5880bcfe57839938c5c099f2b48a1a05716093b52a7a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "1c9f82bebbad8075efd9b66eec44b696f61ed6d017c98cf8adb222625b733173", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "a19efbc9dab510de8efb3705911c70f64449f6316c8160c80de9cd0219f60213", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256u2_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 896, "ad7fdf8f49746f94fff5d64b9d68f191c61bc4cae0d0b065b1962bcb34814718", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128u2_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 896, "fc2171541ce11ba3e54ab4aa736d5df6e3bf89d7555be4c5f29113891aec81fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "17fdceb90800d131a443ed6fd498bdda58f122a202a7038234da08133d739fb9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "26d7e1d82c4c8d567dc1104db758b87a18458c4cbd7bf364a26973e2e08d9ac3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm100f", 512, "ed901495f2e51a6305d9f60e27542b43ae77a1e05d081c9640c1dfd56662b6a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 896, "7e025dcf65085430bba6ffe48ca13721c7c02088bcd14cf4843cb024004801e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 896, "f5d7b4d4fc46ef9d05c0812a32a955893cde13aaa6e92e255b49525332f9c6aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm100f", 512, "2341941fe9c290ec8a00f878462ad9d61785510c4d5cbc661832ede3f6f0cbc3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "fd68b92ab29841c3cfe280bffcc63dc79399554a8e2831c0dba434e94ededc0b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "a3a0f15966fc524596dc59979f374f4d7896c22693436451cd1a545b95d5e032", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm100f", 512, "7b445937a4ff79551642428d26a9121bef3a64ec927599bcdf1d452489442460", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm100f", 512, "861ea6677232df42f87ad245fcb2068c8e1968b7950eddeec524a6f560d7879b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "3c1532d026a2a198a601e9298b528de42749cb2e7057af423af758156d75d0ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256u2_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 896, "40e4930cca1357de84b02f8cee8f04e83319f5089558cb9f1c745f8afaafdcba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128u2_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 896, "2f0f10ab774dd5df8b593dfe842a99424640736e55e667a88c66f45ea2140125", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "b6fc89b2a481e85eeb1a12a410f0c67a897328fef2511996f1099c2a5e6d9550", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW1_lsfbW1_dynB_sm100f", 512, "197690d2f582de7cca5144cd7d6af534b688668c5e8bcfb0ca2ab9cd24dc2bbc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW1_lsfbW1_dynB_sm100f", 512, "88c02cbb4ec5a8b2d863f33200af04480e80f12d316b145bce7f4a18c62dee34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW1_lsfbW1_dynB_sm100f", 512, "eadc64c235a9842510054db6a0a529c3ad176803e202a21acba60a7bc9709c28", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 768, "01a9d9516a347adc5231e290e44ba6a372dcd69353c1f8748e234d34c05004bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 896, "8130ab1df629bc5d90ff2799322cb5a17ffe96ada556ca34d56c4bc8cf3d1785", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 896, "98bdfd92004b25b2d254715df126828d42e786df4c310d6675cea348bf359ad5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 768, "b22e4f320bc9f475a23a737f8fe9e11110b70edfbae2d699e579f8fdc2a081b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW1_lsfbW1_dynB_sm100f", 512, "57c969d7f46113f2deff7f794e3a577069b13cb6473c8db0b71ebf38336f1ab8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128u2_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 896, "ceb2c4b42e3ad20b2994be03b8ae7b137aeee8611f7e5eadbcea17330d81a73a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256u2_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 896, "4f30602e265fca04ac166126673173508c295b03993cf4f69e2dc588e80b1ed0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 768, "cbcf52f58a955cb06df7d8bc5b881932299193b003971a61825d7276eef56577", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 768, "d980d123e71d7a3e0145700d046da1f7884cf8a10eee163a73c91ae44fd354cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "a1a163dbc93fe9e9662e6a9fc23048f4b29bfff6f9c26f4e9f9efd690f120b11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "f9443b5cd0d158b3f44ad220ea557cfc6e23b94d8735c1ca618e9806330dd068", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 896, "2c4989467f74b08dcd69eff2dd10dd83a165bd66acaed1a570b3c912b11cda3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "afb5980d5ceddbee65749466a0b08ab876ec0133406deba423de0231e63c8abd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "f81fdbce9dec156dd5a95072b092d3d74e5f043c26439a3e057febc5f3f252e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 896, "222a812368ffd9acec050ee27f95e8e520bc4b89bb446c1df0b316cdb66376cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256u2_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 896, "bb66e2fc3282d8b48fcf8f65c92b966ac7b81fc42b4544ba95ea3a30adbe4a44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "83fc92a0625b154e16a74ef944fa1a878929d2959f8e1fde837420b1eaeaed9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128u2_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 896, "17a5fa73acc6f7f73b0b6982b87e72c6461ebd2925ae2ebf516d29ac044f0aa8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "196c4bec9af9f316038c2d027b389d2abc36ebce54886dbb73dd800e91765cb0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "de3a3e682c6c1b9eb52e5d84714feea963e64429758aa0aa36c2fdd314af3d3b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "4a0c8c842d25f9648cd697944a667e989c9f9ca3e86686b4f6a836220ffa3322", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 896, "16286ff7dbf44f51e6990abfcea6896b39f859bc18117830186e494f8115f176", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "d0ea9ac1552d13a3b92712bbdb4163e324234b2370b837f2e54073c08a962b48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 896, "269df3c3927dcfe2c4a68db5b945a91a64b3e3b9ea571ae574edde4bf692a879", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "fb7f34d0456fcc946519f297329e7675a88db5eb6daeb204924736f470c2c5df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "47e2a2f2584aa1e0ee39044ac388797c5de1125ce1dfa8c700027808c4811076", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "e5da1c0d004d1f1e6ae78edf81cfebd0958a2f21f7ba669fd9e9761a5972af3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "973d9297915a1147b351083b2284c08342e743cc038fb4072950112f132e6d35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "dccf571b7c9eb15b74fc62c416aae1f81d5349cd08c4627034b76f3160ef35e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "c808a76732e998bf36416aeab8756273ab48e3e63a8c861ce4cee81cc6cbe222", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm100f", 512, "9b4a7cf0316afd4e0a71038fdf03a8c513c0694fb57d9d324c94afb0386027fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128u2_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 896, "393198d8b153aa859e109237fd2d13c7ce73bb29cdeabd34c8866d9a279d0c35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm100f", 512, "d0887fe423fe186861e7c96ea5b76bc4cdd9a9bf23c7abc0c4ffb6783c7115b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256u2_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 896, "db76ee0858e7feadf0dfe077e43e82d426145fca503911a674da8a259baa8adc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "91bd49fd58a96632f331bc678a0023b9a035c9c84f23208f465f47972a9b1e3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 896, "2a11d98947664814cb58fbf2b5444655d2b8d98c52a376e328d14a6eb6d82e3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 896, "85354bac2285dc320aa3a440c2065ed9ab9557dc03539eae194b7163324940e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "80e9de87b4647e5bf50c83f5781164aa67d3dda10422cce2162199612dc267c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "9290bb54cde0c79a143360b1866c7c7edc4fe8c32aa4c0063bec4ee452b1d6da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm100f", 512, "db1f1b3c820129cd659bcd0b05fb317da4486de28affb66036d7a113b7556c5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128u2_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 896, "096e7e1526993161d16e685de04c9b540be69dfb63ff22868158a37207f37af2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256u2_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 896, "4f76eb4846976fd0c15f9b437aad936aeddf323c0b47ccd38ea28b119508d00b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm100f", 512, "f561b8c0ccd4d7a6017ab79081bfc193842f60475ddff5b44686f87f1c2668b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "351f01cfd0d255047677a34e9a64b97d294ac2413a84485fb8b0cc76bb27b25c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "be4685f315d57a7fc30b61c92fb86b40d28ce388e48c13c89cfaa3f6b46596ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW1_lsfbW1_dynB_sm100f", 512, "36bbe74b1b9da176cee99c36d33508b0c6a3fbf35bbfd48311384885f89653f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 896, "0d1049fabf754c2ccd395920ae785827a964129a283fddeb32ac50b6cf36f2af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 896, "22bc5f13d8c6fa4152005ea78e9a4dd0e499bd1d208d05f90f3415438a715dee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW1_lsfbW1_dynB_sm100f", 512, "46d3e755d8db2c4acbd1146dcb46654286aed1af86753e473814365edd5c6e8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 768, "d49c201164d153798e24d05673eda40d45ae3a0f6ed040ef1a043ad39f059b45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 768, "41d1f4ad416bbdb6cbd4421d1ef1195ce2163b1cb1eb40e561742df715a26c88", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256u2_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 896, "d3e095b610a0dff9affef496f90695c1fb2934944b5eb0a7124f1eb69f6a975c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW1_lsfbW1_dynB_sm100f", 512, "6196e325462d2af5cbf4b119e4e9f851c70968b9b08c9f9965ff760202db4dd4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128u2_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 896, "0d27eeab59028692ce3b15565c2506f860693b163af49e97189ac800fabf57ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW1_lsfbW1_dynB_sm100f", 512, "872eb514175f7851063a1de4f479fb38b19c40ab5e8fa58ce2bbaf4f83eef6b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 768, "f776b77901c9d572405e0c49cefad5d3e239f9e43caa8807907a5847962bc140", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 768, "c52257f241d5035b7fc9b09237d35b9eb2543b1042661b25f9ca34ac3c63c0e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "64ba8f486992ac4de445ed80d101eeeb236205b0b0fb8a4f17321ac59a7d4ed3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "12de090529893df132707e23170562023160d71c711edc78c71157468be123e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "6b81966328b19de92e42c772fc20bb196e18efa55d6d5b729613171c9606f171", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "62a368e47e7821e7d3edea682880b013cc9cca0ac42c3e7784eaaba395834fa8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "cf58c8f114fa98bf71353996d764b0d840d4b3479cf7f820dc4dac7beb150861", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "3dc50dbb70589fb4b10f30eaa13bc1b9dc94642d831f3766931c07c576ae2d22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 896, "4241f4db8bb9613f264f2a4600ed9aeebb571324797f893046af72fa120fd69f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 896, "aedc2477188770ef1c03e67add0fe139523bb19763847305d42301f89ab54dbb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128u2_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 896, "0bbf6eee86433624ebdf8ccfed4d7f00fbf37a6826a9bef43043ac57a1eda84f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256u2_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 896, "c9665cb6ba2a5fe67d49673c7483f472fcfc9df648134690bce1f6c59d74f9eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "f2296449d6d6ca454d13f0fb49896d59093a5cb1a7d20c637b8cf63218fefdfb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "a2a7d8f7e7f12124056409322d5cc47de6390a5f2a51d24d4d0f8fe5952209cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 896, "a40807b4f24f120a5a1931c7b5d00029f6489ddaca2beabc7658c3a3f06afe34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 896, "d266613f2a07afb5eb92569bf31568cf85b54eee86ed64abb0a386702b32abe1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "56d42e3ed38618b023edd9c48ef572cc9d70d050de09390d78cf00ab8590841a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "f33b35f24c5cc95b8214f7be049977c313b5116b345b92a3b257bcf5f0553151", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "59281bed3e1c247cb957a2078aa4a67978e66823d374631e2b6836f4239f55a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "92ecad3367cd17b3dcceaa48b39fdf6a59ed3ef74aad9f367e0f60f5c078e7b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm100f", 512, "8ed5e8c57f8a5bc340024213345f4490d3a21891e946f3c5349f0d6385832877", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "4d6854ff2435eb12f8d5f7bb5817688cd9000bd92c5020f6b5f13b2d780b38ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "91160c99f7e13c183cb574656e0202769ef609b92c028cd74d75defe4e0ad2d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm100f", 512, "da6d4f7e409cda12ad07defe01c7de08e1f6f0dc9a529761869e011db09a088f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "0b434ac12f1417adcdcde0015608eb105096f2b38bd57b77ae10fdcf40440a63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "7975a547b7e21a40714d620d5bc8068a0a79f706a56db69373daa8750607bb27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm100f", 512, "ff872f5e243ba72589e50fad91ae9e1f21c2e9d9db7236079d7909de441a813f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm100f", 512, "98a7479d9a253aee493b22bb8f562babfe5a17a577eed7f374d76ae9423e0f2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "01b839b243c56a2524cc72491cb4c2f7e13d37467fc696266c1d0b018f0c28cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "ab6bce29523ee07c1b8b11369c02d9524bd7c9507420b898ddffdfe5388867fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "6e5d342bd5e45536315b06e5f68f2121558a4cc1a9ed5aee0ba24219443edb55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "f80ada3f3563f1309f570cbaf8d9ead9598bda7920557444fa3a319367f1bed9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "47163d481e2d25e7518fe9250b455ecc302a43ae00a50e92c9dad9cb771d1d8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "229196ece6bc17cfa568004d247da604345181a2b0d340610e2b0e7ac6fd7132", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "5886ba90ffc8eb14675f37cc69dd7691f79ed44a72b301fa4f1198951911f5f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "036a57a0ad07e096c11c8ca017ac2f63b733ff806023268420d6897c7e9dfd43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "4f01c0f58f0205533628d0295eae9db6e34beb70082c93b663f552c4cc89f774", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "2886797b70717c3c48c4ce315df2f37c0d326f6332a09a35c7baaf326e083854", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "5b6c6d345e56ed4b65ffcdf1b486cbab7977ebc195a4444ba7a92b84d0fc3233", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "3239d30420925e4c598d76628df92e81c13463713784c527ed4b5878fa5de893", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "ea584dcd65d8c3ee38f76ed3d0125b536f39363edc2898a96413cf1f50d6e452", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "946b8c47884557d1c3b40a7bba4bc1721b41510c6e628d42e023f3cad940f69f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "2799222d1d430099041565e7d5791503ae39e26409cb70036bb59b2a12924efd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm100f", 512, "accd15df432bb7bef19f0cf8136c4f9034f3c86875e250d058d44d5813887b37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "3aa16412007f19ba712469fbf70d92ba2ada162a258156e0fefbc88a09f10d08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "a65f6bb67e10a5e91ba31e8641082b34327bb06bc86f9a84deb112dc9ffa8658", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm100f", 512, "3df665a6cb23db27960ff1ee39b39d8ef6263ae4bd4be0c9fb8c3814ab282c90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm100f", 512, "7a21a743066a7ffadf93cb3194b4a14e5258b2bc1228c3f299aba0085b162bed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "5693fb8536554256199c1953f3854e2044958324b6d6ef5b4d0c21ecf77da71e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 768, "e42e665aaeb67ad21f347201146cf16771028a0d2d00e60064366b7e7eae04ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm100f", 512, "d84f35ca8d00ba8b8a8cce9b7af5f4d6b894c5c707e486f2b3ecabbda53ca371", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm100f", 512, "1ac8ad3930613ef85c616264b2240075d1292f1cbc0a30ea272ff15078a0dfba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "2eeb91ddbddf1d80c86328e871ed99ecc31fa083cea39d3cb1779a8f33559ddd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "061492d97cd65d539eefdf76557c89c2cb2c7f6646d82fd33c833ec806ce2fa8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "0d8ee2e47dc92e79ba10660965d347b7a2da5711ffe32ab4ec87ba24a3de2265", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "a223828e1833d92ddab838939bd08bee47f0df13a56a2b53055f57c09d6ecc26", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "dda7926d0f5d09b20557ac3ff5a69127168a9279b14018b0590579bd4df90807", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 768, "882615206f235d967e6d62a1ef91b500cb29487f460eeb1abbe36cea0a62caf9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "58eef56803ec5abd0c9b7ec53a59180026e7f830296bf196b659ff0894c0353c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm100f", 512, "03a04f7315163b61959dfad4ccc2eb7218fd3043b679362283f403c436fcb1a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "f3a3c0b13385b893ab39ae6e34a91f9662d8152f4e8b07d371b2c69246251e73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "9873ef66afd1b39673af89fb40f8fdec28bef412bb0ebf61a658e1bfc1028156", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "990efaf083a343e7a5531b3b87837f9365d20e35ee20ed7ac8bffec797b84d62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "488d74895ef8caad3918ee333a4b94a8ba466024107fd5bea7152d4a3cec2386", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "cb519536c568e1d8671c54c50c7b9871a95c6412002c18179abb79a428abb919", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "ba5d562080462c42468877fdf47e90469558a13246754669a72fe4512eede729", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "e36b7f9f90a0d2c9ea96692cd384a46a526040390042a58ce9c20ecf27c948e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 768, "012874a1eb43400766ff9780995ed781c9d6083016df404afb50edd48bbfe8b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "4b0428c544ca27d633a5fd7e4fba2643afd54d5cade30dbffb99c75e5e9aef38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "8536603a5bcd308f13a235aedb933103edabe6d1cd80655ddc9da1495fb01c3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "07bbbc1b7f250c9b0588fc14b40ad060cad6e1d40c5dc715fac027f479803e00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "92867aa4b3f84188350d7edd9fb0e85896a5aa6cba7a1156b539cffa84ae7c1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm100f", 384, "07b73111460aeb27ffba936789b39d00121b12f119afa93d094f68ccc2572c1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm100f", 384, "4df54e9f36e81c58dc7572f41912c1a2faea16944b7814a93c371ae842340758", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm100f", 384, "3deca4cbb4b12b13426c069d21ae07ab575ab0e6558cb5f677de42b722867c00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm100f", 384, "ae6dd22bd8df63addcbf1e9305db612126558933f3abfd5a9d2cae8949c9181f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "e38451234be54adc3cdff58bb785ce064ffb520e0c03b56f8d160196bb3edc94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "91ad4f8b8051855179ee7d56ef54b1dc5fabc0f54b4137831c172a42facc5768", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "d8cea552696f33742014f54028e0d613e5a23293e7a1a053b6473badc2b01a6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "863718c553e3f343c864d78fdcf1a2198d9d7d4af8bd308ee5a2280c403f2f83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "555fb35a690b10cd7108510bb92593e880c1ea9a5871fd339717939a1ac1d3db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "ec0afb212675964c1d5cc5796b2d7fb475ff6272baa11ce5d1de6420e485e3c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "81ad997e83cdec2286a1db8eec9f5c93647c38af89c426478a9557bc0b874872", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "135bd09f5f51520d2850364ccac1149d6ed782edca9f71bd7b4ea4839044a919", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "f3f5f48094cadc616a1f5647af7bbc53d5d4edab5be506afa994bdf031a14461", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "4359010df5b384b3f337abe0b6072e52c6189082685bef28173e76515d2183b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "04390ba3810f222d7b01181175de4997885b56d3a2796fb45ebc1e7513d51986", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "0fb8f849663b1c4ebc02e2263acf927938c09616c5ceceaee66474810f3d7b9d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm100f", 384, "85bc572aa214d5cfc95f123a7d4bf409cae205bbf26a21b2e410ce6dc667e252", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm100f", 384, "d652e99cb026d843311483ddbfcaf4fd1c2763e45edc028805ccb3d6cc68cadc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "b354d5e26e2bda75d7aeb904f3ab5c5ff1178dd19344533c606fa3cb3dbde0ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "8dd9aa68319284a8470ec49a9f523816e183ddac50000c16ca4f766514c770a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm100f", 384, "631170922451ebbf3622bcf5f5effef022c4213ae567df4c4a171dc65cedd80d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm100f", 384, "713377d5048bde86a934b6204e4423cd148293e7b5641e521395c6ba7c878a9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "833096cc64d6f2b75ee0a8824168ec993c236d4b308173a1c06f081a685f858c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm100f", 384, "92d60a48860e4d519a950f8b784047883da8f042ad90e38b91457520d3c0304e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "07282445b1b1e4dd7b9b2002d5aaaaae98999c7f6ff5d7172aade68351f887b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "37bc964346e22b7f05b60dfd7a9f7dcaa813a5a96fd3b721577622e1145cf443", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "f84e6f40415387317781017a9a1ada6c146dda240ca93d660f05fd39cf77b7ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "45dbb0faaef582230a97f3e82a56b1dc15038dc02ab7fc7dff0a2fc9f45f2c7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "e45d59c04d5a9c58c8e1c2c70f55f0444d8e6a434cae2d84f42454d0ebfb7d82", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "825705d83d7f28be177b8f3f1e9c0d751003df70885bbcdc6064a834918735d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "c0812896a5e401f04a0d87c5e5030b2e0f7788e8e64b2606426f763e05473e0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "cc5c6fdb8f783785ef73a033437eb8ded512b3d3cc34ee1e546cae192e1105dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "851e0105ced430fdc13af0c1ec1153c60136b5dfc12e7c5b1a5a5d68164e4971", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "21e719f80d49ea498fbc1d3ca5d243ff9c13ce9b73f7a5d1bb7649a02d3913cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "07fe557dec2d10e455845a6b8ef2c2aaac27f112c7b8ba1e8ca88fc50e313daf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm100f", 384, "44d5e6196536e265630a88c4681d8b63af12b1d49ee8ef2fcdc1be832243e18c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "32974ef6d021de536ccccfb215f46ca0650eb7c49fcfaa4db3ce7fcc830851d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "454e32f438a1ae98fa094effeccbec5b764e75faecee7be03aaff1c232dd9053", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "eee6b6b7f07a0dc2deb1f635133bbdaf1e38a1dbe8ed315b633a46767aee3298", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm100f", 384, "81729b1e97da0a112b7db41ad2de35f609ffbbb319ea40cc11aecc7be3e60402", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "d83971456e144accbb3e19d2da124078dd093b0a9642821cedcb8b584efa5a3f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "323a58df590f7216aa31a4738727e932ad5edfc4c187c2fe178640b99dd1e83b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 640, "e3ebcf2119ac655b79ad31e88f45988f1de08e7cd84b01c2de9ba3372e0fb96f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 640, "8bb5abe328756e224429496b905f92c68f7e86674a47da693a22e9379a3883fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "2d0a6febf0faa8c3346bbb44ad17f06803bf0eda0db663899058de5ddc5bf220", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "a72f878bd6efd6405edf1b8786cac5821a773bdd93fdcd668a3401a10bb96af8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 640, "225f9f10c6f762fb2c7010f5f9698d39b004bedaf998bc8745fbecff02301ed9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 640, "e2cad56bb0c08b0e836b941cdfba5e21ea66efae3c36b3fda6b5fee268e6d44d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "6b088a9830e33dd950e33f4db58d69a7f7759dc48f9e7833470e561a7bc8bf65", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "2632d487c7df84f48a04d2bca91bf462ef3d4f875c3878474e157143d9687d1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "fdabe3c724a070e512a62ee6a09f70fb716a0eff82744b072d8f57c93022519f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 640, "9682f005e672fb363aba28517660fb2cd9a455ac270dd11e0b02996826bf770a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "3bc71ac678fb272d320a41efe13a9e76c5ff6e5ab38af886900e9ac350a0881c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 640, "1584598e3aba557333b7fec9acc307c4bd9084c4a6518fc82d53830868cc121e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 640, "c3c84b7ea17b53de8a29734aa619a2e48a39416132d2508934a72be021be6c07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm100f", 640, "8413e9d6a4ca3de609c2817d1f56994de6b58078fafc0e37f960dfe3b6a7b1be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "e4ba3748bc7423fdb6f2b89b03e2bab1603d05b72ee417a06f02d5fc08786133", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ee6a9c43c5cea231d219b163d7a8f9db52cbdc74af648b454813136c1b5a8a0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "2ec8ce4fd009f6c99872fa470e4c50da310b3007e06e0d5fa2684ea54950653e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115216, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6a67a132bf699805362ce588ef7f3f2a8a1a86f186579c1857935573314cdde5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "674622d6ab0bc89b60de9e841995c69f0279b69c209b5b02d9ac37171afad297", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115344, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6baf212355188195320c21bb847e2c06e7b3ad289711624b841cc9af958e9ab5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151056, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "27d3fec513e9e020e6fb216da306f9d81cb1547985200e5b96f4df6653a9d32c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "bdaabc5901622ca6937c10ec87329e9e59272ae6a47a5482829e1b834e78b780", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "2782139e7a35fb6653e2f1629da25c0b37e5aa0feed9f8c4547e9527f55405cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187280, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "33bd3deba905f3c4c8ac5be5310dc163c92906bc064336df6885f752663e0966", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222992, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "d3a5106f45913cf3550607df4a0395ecb9663d573ae8ab091399b35521aaac01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223120, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "443a88770ea61b003ef92c55401b7f80f1c657137514fe1cdeecfdf9037e268a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123408, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "230717950af3e666e01be3287fc6a9c59d037cc42c3c51ff7deb4d46a1a34da7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123536, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "e49390db2a7a742238bbd1b7f8e7596bc720995168a9fa09504eb5bb3f463228", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161296, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "23bbac5f050e4e8aa1b694a45e5a637067537340815506f85d0804a2d8ebfe10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "79d4d28448d9d2d46473ed4a8bc6409f6e99c837da987739579e4a6d0a9a8a53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161424, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "3805f49e0b2a3abf01dda9bc8a3c3da2726f041084b57f619f44168cb4b7c430", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 199568, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a3ad301844eb811e8b7116b3d96a13f87979c67cbe97d061d5785b1de38e285a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117272, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "eaa7de2b4e2c5f9ea5fc51c0b21a79a125363ad68c22e862f4f4926f7cdcd7df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117400, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ab75b783cf3e8dbce5c20691a3f851b64339a3944a991fcb79d665c1a89ef52a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s2x2x2x2x1x2_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "1b9e0c9e764a7fe901387d0a6b8b344f6f0b141e11d7307aa3b9ac6b0a41b513", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s2x2x2x2x1x2_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "60cc67214c6ff253419a9c3fa75e240a653b6b0b96a4d14800fde078ea685e53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "8febf75ef2160a8fc1ba91f2f3901c44a22dbe0720d4a6af63089a1ae2762098", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm100f", 640, "eba72cb44c5ed5347af55186ed20eab82bae1bf4ec391f91a7bb704e3726c506", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s2x2x2x2x1x2_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "878d362caf79df3feb1e9231d9ee1ace2c9a76025ad4af0e1f4247dc6f4deaa5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153112, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "77c7dc27df9dda4c04376b187a812e7d1a44856cc2513ca06a7e4a2ba242719d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s2x2x2x2x1x2_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 640, "f7f6cd2200790827e110dddd7a1cc2e56b58e30f6e1cb8e8b1e67b1974747e46", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188952, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "9305929dd339f601c467748d34eabadc7dad33857d0b56c51cff9c90438aab54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "d373a0a57eb4e79fd667aabbda7d1fe7b509b16eccb7971fd78df0b62af54f11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s2x2x2x2x1x2_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 640, "311a415e18809b4a9f6738ec70df02da064b9c6fd5ab5117b7f6fe3cec63c964", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153240, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "0f702558930bafd2cf9bb7490e2c5fdbb094939a33a9080dcd49d805274c6e0f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s2x2x2x2x1x2_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "5bb47030062176304bffeedd4dacccf9aa9ced79b49b4bb53f245451e9ed86f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 640, "77dda1a47fe7e0f386858dd639919d3a66a6063ae41284c2e9bd0533d32c19d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189336, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "f652d73ae2f4d24edb73e9a4528fed1a317b6e900b16039a4ad10a2c2788c195", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225048, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "42e9ccc6a0ebd9e73b5e358f618cd840b39b83d40510c9e1434cd82c09376b43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm100f", 640, "a381e1922dc7e6a07bfd7238f575ad028a2470239e5afb900f3dd973b7778737", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm100f", 640, "388dc59de1d29608007012c5b798e4e696a61450a87f961cc4648537ef4c8576", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225176, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "e5f19596c4c361f9368533118f483221a6692f3a5cb746cb203b43a72c7d2b90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139792, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ed403b327013c97df59d7e2343df9a2f6d1460746e97f86a8e4bc82209ac6d9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 181776, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "20684e6e2db030f6ebf7746f74054daadf92ddfa00e1d90d0f102fe4ce83d14e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139920, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ee5f7c85559e8567fbf777af242faf14c046bd95e31df83b76949cef78a38508", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 224784, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "16aeb113e80e205b57a87b03cc6ca6f2103e4455edb0cb1e9f9a90b3a23579a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 181904, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7fa0e8fcda123ec4b247838af730161c349f70ed5db5e8058f97fda9ab9f8e48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225168, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "341e8ce14bd77027aa6bf09a900ddaff9fdb875825c974d871cbfb701d731b9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 127512, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "0b52293f2952caa6f52bd0e40b64ff8f9e1de34da1492e19da5837211093e1ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 165400, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "667a0ed43452e985256f9ee15852d9966b5aa7b244801e5985a3ad89dab0bc1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 127640, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7837e4da820ded6b5ec252ac201588f402840d7ee6198dda30760d3bbb901a91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204312, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "744ccb7eb4aa148a1ecc66b58265f119475df06febf458f4697fb7c73cee5bbc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 165528, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a95976bc26dceddc0f52b0b29d2d1f0d77aa79724d422af900f7065a44888139", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a11a76c652f6ff642a4b5da78f271cdccb0b8f2a0390dd0c87f7eba9b26678b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204696, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "98bd16a12c6bd65a49193562a07f344844c2bacb62227e113d4f3596ee1dc3be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b566d872dbc1ecc614e726d2586dc7c7a19a541b205d5562cd0821cd4ad32322", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ce6213b7936d54d971fc512b623e44e051d04d3543fe380a761c2c3f6cb0a919", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "47e257fb2b81bce764bf29164d9132b6946796a34a428e01165a263ec556259e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6e8744a6c20a032bcfebeef3957f69d964a2b9f0bc9794d5e4c5ebcd9ed978cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "1c36a20335e5d9bc49d12028c35f527853263822941ff975e9482a5d2f518a80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b7939d95bc40c5ea4b709f7f53abe7bf057289d83e83244f55266bbd40764717", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "f12baa8bbb9d2928dae6abf6525b18e31ffd007c74a9ced26647c75b11181308", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "30fcf2bdb90b075a6a7e9718fd3275277ea4b3d8c5149094463b9733f94831c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "c6cca31d75056fdf73af9816d380490883b8455562685edf649e65d001ef9063", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "1834cea248644e1c2ded277088489f3ac570dff2e1735f63ba9cab9c4c89a385", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115216, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "df3a890a80b9cd03b41bea9193d9833929cfb5b1ccfbb2d6bfc247cc49fefba5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151056, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "f9a524d3aa9647f84232a0cd504e3b6b89c8a24d854865c8af51e3089630baa9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "00f6e17878d9100cb5a06c6ebb80380dc8d171235826e63bf2c0acc3401454b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "635508e4469f58da589b959962dc69de5b4dfe48efd4151bd534399be4d99bd9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115344, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6e6f254b67050ce49988def8a2bc1160859d78e4ced8f1d24c9023e8dfed9569", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222992, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "20cf3ed043c3d36d99b691fbbba139525a1d3daff0fe794cf1ea4413d21f654a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "08cc0deff35663a8948efa5142f951335dd18692baa43643934e097bab472d10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187280, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b8818885ef7952039a53284c216a45486aae500a39dcc7c40bc3da701825446a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223120, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "8409bb69b158f4fac8f2f45785747a4dea3c0b5644ae28708ef42f3cd440b046", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123408, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "e7edeceab64bdf03c07a3fb097d05c237a51c2b169c8d0c1ce4a5b05ca6e835d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161296, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b8edffd9e9f5b0c48a16be9341f42213c903611395b7a6c991562420bcb3f1c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123536, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "484933da52b1f0992731ca362544f8fa22f1b6b56b4a5aaec3070714e08a336d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "00b8c0b65663f5a7706a92abda1d71e62a7b68ae4d91bad78233a75cb438636d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161424, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "d5f14e5dfcb674461b31c71527e594a8976f4cc2baa64a8aebe88d12b3e6b03d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 199568, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "3dafa0837108f542d5a3c4cb57ff70ad446a746b19d07535545ae652873da87a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117272, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "239e3f91dc89abc7439a47e74f21b31054e4cda2b6070366450db0b97f88e245", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153112, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "2f9ea9089725e12c3ab6d4e8c3c47452c79ad09ff79b86378ed8d0e87fb519f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188952, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "e5fa6008d39f78d0219d96d10385b02985b32bca9451df8af2aed3f314b24675", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117400, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "d556726276e94aef962faebce1d80df9e1ec21e7edabed062396a8514117ab9d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153240, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "04092d44af81b03c6590aaf2fed09176eda13338b44b0b49138e24244473532e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225048, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "bd090b0f235d7ac17f241a983b4a7de00a5f99a2245e1c172f8001db8de8d4a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189336, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "2b5201f588c7c4c73b9137f8e53e5948152ee75bd67eead0687595d1905ec75c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139792, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "393a97460253b15967e3f7911f9387380b9a563183366a2a4cb9ccd3d4ccc80c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225176, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7fd46c849f49018ed59f10cc73b5d5ef2991a1bd3a3e4c820d1d9b7b227b7934", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139920, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "de07b9f3f47ce0ee24dde2f604a8ec6c442d7f43344afc2606c032338dc83e5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 181776, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5c165967e631250fba6c6a779803de9ac2453ceee76a72dec1fc9c9ac8a739ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 224784, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "f357cff6746844389694ba2298b4b93990fe3a5cd1905dffd7fd198f8a3d01cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 181904, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "059be47c957b106d8452d1bd6a3be21eac5edc865e99d9dda62684e188e6c855", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225168, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "70b25d2e35b5ac6910d61c907a3c7088e4044c4efea33956433bedf6f4e555be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 127512, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "bfcfbef0255c5716ac8b99c2ec4e751b46f9e4e42df9dc3ce2da5e723a9f5e71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204312, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "d951bd70489f59097bd26ac449e51d8787959f49f5ba170718b458e2152c9550", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 165400, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a9d89c7482478586879b14d37af4e43bfb4543992cb2749d22d503a2b25be069", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 127640, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "46eb71d072c90a723a75e4982fb26b431c89353556d102231c448d35cd4d501f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 165528, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "49ce8ade0ae7ae4eb400adc777060bda8505d7bc5e9b3bbaccab6310a733282f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a472c96bc170716d51bc6e068df6703a55622ae07bf512d5fc45cabf18e2a775", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "f7d4e669e4ad254bc6119f574177472e254df92de1d8ecb031020c9d8631ff6d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "3cc0180e0be158ba061dab6b9a577c950f8bf857009a24c9fef729ea399fe484", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204696, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "c2aeb47d127895c6fad7775f22b18e8f9987d0c6d5e776cbfc5248f39e28f804", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "0d61249dfec300914c2693c2f9b819768ac801023529244c29b52ff1ed54b914", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6aead3829e29689e4b78283f8e2f3c40502b1138f91227680e49f43866125bcd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "de3c687d1b5ec1660d6691151a92f2a73b1dbfd63ab32cac733dadcb5b2de0b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5c98c726eb59a8327b377227b52563cd9584517d72de37fe392aa841cec1ace0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "dc48d03ebd7b406ec0fdac5ade1bfe55faeea739b0abf8616246e922d7991737", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6a99611df12fc0ba346163b32d290097561f90a356ba0c157663e30e8b12b22d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "314dc2621325dfdca0258bd4c6dd3bb80737191a7a42d962b7d32e13d51dd9eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7d14d78cd1dba286c2e58666ec80fc436abdb14f8b55538e7e7798ca6d9f235c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115216, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "79719398721006ac8e39ebbf8a3dec86f31cbda9a64e84307fbd6fce5a2a6ee0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "c267959e239c1102e24123be954bcc456818ee54ed0987895ee47c679ab17707", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151056, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "cbaf8eebeb86937737a162236a28d7d0f56d2e10ffd9f144a5f78bc63135c3d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115344, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "441d26d66e30c7302eb2126a50ef24e738fac76427af4ed253d95fc394e83786", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a013f0fd089e1e7201e7e2151f52b477ba267b3b955e398f3e4a192e1ff6cf03", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "8217af3bd1ba6e9551934e1fb52e143408eacf7d886144dbcb9db308c589c6ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222992, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "bf227ff856c52e3823a573e453bdc1c7a39c00c0b77b0b5b0d5a8eed3a46221a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187280, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "120ea4fe65b2085c0125afca03c28f0d5ddd20a2e72ea181a81e04d4aae67f7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123408, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "17ee386902179aec8b963fa7a0779871167a2334d8a880a26a7f968aa474c620", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223120, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "cacd0e6c727e78e5ba180b314648bc5bf1acaedd67a4acae0cb5b9cfda03d413", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161296, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ba1ebc2c08361bfec2bf4a9ceb9b63bd1c7d5d4da981a3e97d88dd20b2dbab92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123536, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "523c2ecdf5bb182b6344009e9cc7f1250f1da747fb2ac07f012b1bc3ac691fe3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "96ba14d847a3207faa0ec38e08a46402fa1ceb37f4e2da62140fa1296d119487", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161424, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "c812f3e2ee93a1d3a656b66bba1c9c7ebf217c15b2a45a086135c29c6bbf2aa8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117272, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "458d1868d7e6246b0a8607279aa94b19f4944b5c4b9a911516e75a93e1e65608", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 199568, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "8646615b34a98dad1fe2e624827d3aae684e7ea944eecbbd3e5e3701cda20722", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153112, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "9783fe6cabe031e0bcfab95d41a7c19eaf84e68ce19419da4a0509e54943bb71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117400, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5651a82d341e7b39f0e6eb9ce683eda01d41017149b51f06efc624ea6084ef2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225048, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "1e2b3407aef1172ed3108645d0a0fb628a359df0d2ab01498dde930c7862482f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188952, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "1c47f5d21060b7d2c05ea0a4f215a6c03dc3122c4148c0c02b2ba8513c7a620b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139792, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5875faf5479ce5d9e3935f835e0dcd3acaadbda01c45df5b28ebfc012286a777", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153240, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "51004486540f5fd148375df39f8bd56324a16a51504a0dec553ab5982a1d9e79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 181776, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "e492cf0d7fb1e44e326674c56ee6a4c980af5c6b2460c7fd366885c5de486885", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189336, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "0cdb33c83fb32000b1646454fb3e6be22878c9dcc28b9dd98f42394cb49790f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225176, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "26073c7400635e63b003bcebab8f2f4dbf00a7b003fe284c2abd45ce7d5ebf21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139920, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5d47ce71eeedf604d5ac42749a86b498b4205ddfa92f8bf52ecaf18c954c13a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 224784, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "631932061106a3fbd35775b7e014aa89a30508d68ff65fa58e4ec332229c98d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 181904, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "3a15fd6064844ea2d233896c310c840862053372b7d89e8e0e1e2326bb9c93df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225168, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "0c06de75201c3db361840ae5c9dc839cd578acdd991eb2125b551f685481bb5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 127512, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "39feca481d6a64ebf43be6eef4253f55627917f029c56e022b8797936851791f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 165400, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "cda01259164114ba8207a65cd4cdfc269368ec3b435a32e6e88e9e993b48a7ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 127640, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "1411fa3dadff26ff9721e48d3c10b5b0d5e75f9b0f8fba5c4ebd6be043002256", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204312, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b16ea8ff629bad066c127ed13c559a8eddd82479ad411b8a0e1e4b118ab9ce3b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 165528, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "535579d9c783da025d7645c08978f354f0176e7a92d97c5c19c11bad1c1a6704", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6934bc4d03f307a9105133482af6043370573f637f2aa690dccbfc8212c73516", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204696, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "da133fc3726f3587d9ec07388781bfc4ca0df4b89e228003dc55c03515375330", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ece55b6d5cf458f716a29653eabad7a7e239baf869e3def6c3714e4f605a95f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "523fcf511aa0ef822d7bf60d0263b0880fb41d3f95a7d6028d6d011293b68224", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5c6826f25ef7d1033ad7499649c032370e1bbf17129b2c2a52d4a43bc1cfe0aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "0846ab2ac3e9c23d4aead0a06b9d187a968a7e4f337fb9270de5a5d84cf3bd87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "402391b7ad94515cb6204dbfcbe90f475e1b3c69de44eeacb689ad845177a6b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "2ccc45caf7296411be39e991db044f1faba8029227f355f308a7d24dc4e1917f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "409801b7e1abe5037cadcdfd8cfe80061d7057b06fca671e42e84653f0489a65", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "fb7948afdb06760740e69376c79e6a7d99d535e54168b47b7c76990d2ef2348d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "9205bc501ebf70928f92ef890ca8415501d9340cfd261fdcc5c6a61e89558335", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220696, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7bb3df8d7665e6a5266d39f459e61235086cc05c20fe2809875d3738fff3b741", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115216, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "2f4d3dd60cc91ce901d73953a4aaa43aadc8bb737fccba7fc32c881bb9af0361", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115344, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "319bd59a773d4f3b5bd920acb1b6c5631b58a92bed259dcc755dcc5fc58d732e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151056, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "4d63271ee518cd11739e00dac294a1b06f03775c3f9f71b3d7a1a59bba4be568", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229016, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "25162a65b753b7a45e6d5452635afb3251f943ce3173c53f94244ca5fe28e9f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7967036dac366ca0f86ed260401a98ac0eb3ef3fcf9bd0645ebfe918d5594339", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "1db94c450a36374f3a24cd34aec81ef20d1cf5f80dba4cd1a6a93da69a1ec788", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 222992, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "0f81f9644eba6515d8bf11e6d9fc08bd78bc560b8c72ce5981975424d5947c5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 187280, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6106e1cc96513ac858f98a75664f1cfd875f623cf2a653d21f18ac1227ab185e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123408, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "2ca682c0c3135d9a10c12e2a44d3a7291d8dacab012cc92d7487dc0d6ac588e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161296, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ca27d08eaf3989df20957ed7f6ac62d0513e234bed9c4aa1ff18522bf4b063a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 223120, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b954af9a2245d39b50ad4d16ee769f06f3374ce2a88637e7a1025ea44cbe4d71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123536, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "dab58890e283f89823736c4c0a9babeab1e7bc4866b46d692de7701f28bb1cf4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "b03e74b267924f6c9b85f4ccf7231f93076a55626c2793ff0f765259bb8a60ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 161424, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "06e32f5bc3456eb147d2e721894abdae1dba938f105933c2f3e0404e6c0aa29f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117272, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "569f8fa6f795ee8096c929cb3ef48382215465c17d2b227852dc720d733e7138", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 199568, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "83fcc5fb869b1c41b8f5ac3cca7ad36631fd1a65adf05370941f885e8745061e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153112, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "58f9315533887503e9ad370b51518221a6b57176abfb4d9f9519a3c1ff356c44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117400, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "23c3f2750963c56f91423fff21f0c6afc870285537dc0748538f0cdde20a4b32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 188952, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6f2d491d8580d14986102e44e756f300efb1e0a9b3e5e7b32ba13f6825dce278", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 153240, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "10e06532602eaa5c80e1012b1e1267df600365dc52390a6412b6a002866e4b58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225048, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "1f1ea5bfe1a6dd5e237651dcb54eb28c6bc829954127aa159aad886534359c1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189336, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a84efd46aec1206cc0f7db4b50563f5e1fc1200c26557aeb285ddd91fecacff0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139792, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "19abc9e0eadff0f22d9f07331cbb103543e43d1cbef9eefc54b61fad4fbc309a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225176, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6863e789e14de4d5de6eb8ffe6efaebcb2293f02a01570a803c952beb8b8c59d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 139920, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5e982dbf64945f51351b0107dfc35a76dc89e54b6e42a4319d0521cac11a7b29", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 181776, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "e1e0fb96a5244eb1f9fc87ad6d0495e1dbb90ac1cd0d5b7eb819e837ae100a89", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 181904, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "59fcedfa43861f100d2cf7673bef15f2a85cb56d92cbc16f7db106bcae227c16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 224784, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7a62b0ec881355b360f16e1dd31d8c7c46f38fcaff0bc7041bec395d45750743", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225168, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "e53cb1ee3a8906c2f2fec48aa0a0ec7257728f80ca6b53f34188b51a995cfcf2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 127512, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "9224c91ab1689720ec3ab7f2e734d80abf42c17e6afb75b661cbd66eb03da817", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 165400, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "1575b6575bccc47b272af1c2e9c2c52f420242387dc830236e6850086c51a103", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 127640, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "043f95d77a8235e4d3636535c385ab39ebc302ac7761c73803d1652cf21b5138", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 165528, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "6054a830c5bcb5b50b11bfd9e4e37ce9a572704f6fe82af533e189709fc85d03", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204312, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "5950c7c02d76e11cbb09b0b508a20dfcaab18de23d50bc8e457e4461f1ba62d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113176, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7def3e2a463ef6c03d24301354e3dfee44ea9a3252fd667e5443273e7ff53dba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149016, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "623493451004b2b878ef2d5bd5da8e958228c63a04997722e6ba044bc04d3e04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204696, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a75fc68c9a3f02706f55d19bd4520b804f2ac4b5b90704ab3072c91b231e2bb6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 121496, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "18957b84d18eef97e623398b304681fdf4c67930ca0d8fa026b87196a7c07ceb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184856, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "e054508e1c6e3bc4b46afabedb833bc4013f864d0e2db8fddb4845479266de6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 220952, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "65a6d35e3a76bb5099de7d2caf2292b1cb55dce1d6a5a1dc125047a5845e29d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157336, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "42a9558bc2b0cc05cccb60daf4aad1c5b19b42898c6d971fd2f069e07ce9d274", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193432, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "07388140434f5459f8bd14c3ab2ba2254e8c58d283dc58d9e9163df4f510cb10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229272, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "75ff5265397200838894532d237e2c06039aca0f5d4a135b572b49f7c12c2fb8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "8605eb0795d4dd17d14253f6cb889d846b2c7d5261d76496e25b987c42e1d3bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "adbb9704921661b0365ffb70d4bf365682bf1efb1e3369da91422ee79bb4d341", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227992, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x128x256u2_s4_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "81201b0320b9d8fc5e30292121dfc2b37dae04d794b8ee8cdeff43fbc059cb22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x128x128u2_s7_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "60a29897ab7ac84bd6322a234a6204c80ac95a3e0049352b9cb5b031588cdfa7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204440, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x192x256u2_s3_et128x64_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "8879c446d07b4ef8443c176ce44ae07c80243134250e0fce2eb963a3e39f0f31", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "cf8a6c09d770151127a01eba550aab04594a4275de1247619cf784b13bdcc837", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 175000, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x192x128u2_s5_et128x64_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ccd6ed7e01147b6b9caaf6346ad8b857c2e9c5150c2e24c0bff51720bc25becd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "12bfad8191b86bc17e5deb2dc97ab67b4e84c6fb4631899070f4e05f34584ddf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227992, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x128x256_s4_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "e7d2a25288cdafb758fd4dd8347b784e5a541e18f3b4d85b0c118dbfddfca008", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x128x128_s7_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "6d38e04ba3f6138c63a3cdde8d64a0fed832eb3580939cfd7b3eb9bf47d2d286", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 175000, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x192x128_s5_et128x64_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "e944f23508154de877bbb715e506ad63031fbef3952654b552f7849f253cbad0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204440, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x192x256_s3_et128x64_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "61826136a184c69b927c2291ee9677dcf46ae4a836275cc796998cb7da340cee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "4c87fb1f07bf232f77b85ceb4cc33a8af47edc7cf5d9eec2fc7d51833c812cbb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "85548ab02f6394d09545fef9427160c8f7f0c4bdfdf181b2521d45e3aa10b2d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227992, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x128x256u2_s4_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "0289f2cd5b2f0630c034cbaca70189c08f3476b58eb5a1001b66e0d9e5232114", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x128x128u2_s7_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "a50930caaf7abefa170e6c4416f6de39fb0191827e94071b2da4e69da8657dc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204440, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x192x256u2_s3_et128x64_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "74de820f665a734fe19e897ee27ae3faa32c75b25213c5798dfe0a8eaf864316", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192152, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "033fe7a7c1e6de214260d347441c456b4857b265a8c6cb6067be5261c332951a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 175000, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x192x128u2_s5_et128x64_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "16c41ff5a4d8ac47c20e25529cf3bd9a42c6423e1ef564a5ff25100f09f60c1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "9333c25aa4ccd2b12e223650ed85724f61a655fdf730d1f12f0d8e7bf92ca2a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 227992, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x128x256_s4_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "a0cf05205f80c4f8da1c65ae1e42727810a5645aee409b97bbbbdc8d19ade7cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 203672, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x128x128_s7_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "3a81404d50098ea0692b04d313f1a0dbe79bbc538846d37dca4813f1954bf61d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204440, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x192x256_s3_et128x64_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "d61e5744d39936e1f14f769ecf106514ef73fb6bd59548a883880dfa0af8a3a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 175000, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x192x128_s5_et128x64_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "470ab80008e3db1b54c7b9fffda78ccfa500d76d6d41589d7777fcd3f7d6d196", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197272, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x256x128_s5x5x5x5x1x1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm100f", 384, "2544e62246086b55d83342d6cfbfc028856b1399795da126411b28fff024dc9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 197272, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x256x128_s5x5x5x5x1x1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm100f", 384, "953882513403662b6e8d81cf7ea5593d2c073d905bfecf9a2e8076cf0dd9c92c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "49d979eadc33f7cab8f3f670f795440a842ccc76819a26304ce676cc4e91bb4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "183130d359b4e38cba9f43ee29bc927654e1aa409de5d6affda6d572b96b3c43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "17236b75faf8863f5d1e0e7046e32ce504167fb057325a25b2e0ced88925f132", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "2e43225aa2bbcf96448a1869d4a3e466af86e94ef6c01451783fafdc34da9a73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e7ec8aae6aade88f49a08b7725325f0d8a2ecd9f389ba1c90017c71caec67427", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e421588d4e34f19952a06cfb012c02e76eb576b06b3222525ebea5ca0f3767ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "4278e6bd4835ee5312b3900d789d7df1c9e448466a840f03f4197b5401152ff5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 72984, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "1de0df0a19071c667598705e2ba6d4e2b25b0d3971106adb13f36a41c47cf444", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172688, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "258a3b50ed4db7418bc1980510c9cc8ff2afaca42d40a9da0397558525e9ec57", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 73368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "2d686ad372d6f5a805a78c72497a895dd53e45b7527e6baf9ae70380a2448377", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f521e66b9267fb7d96970a71c8a0643536c5714d1a6ae71690b6f0137a33deeb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "619531321658866be0faf708c8f59002a1a035bee94e4488889718ecb5c81517", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "079e84222e3058fc720908b72ddfb69b89b2158719ce54ffcbab68e173696e79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "b77ae9d83caaa29bb783ab0b2fa2aa55b5b697b2290f5507ab63b9c2299f3ad2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "bc3731e16aa2a39b741f992793faba0d911fe271eaaec8b05faba96de907cb73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "9cf170bc00d02e07846fa28182bdbb4cb640c280f1698dfd2acdf35ebda4652c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "971c9c5a3f7980936c7ebb218b44892ea014fab6659adce8a8370341a0f60537", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "12c37b5f9612109a13da1d9ef2709261612acc73243f71f39a6beeb383d5e376", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "4105e265e6e7a257b8a3c12f9502d42b4f4a590835bc31e612840ecd1c047446", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "87083f26cf3e3a08b4679f2b2b35ed5ccc80df743e50f9d669590787a3a663ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d3be91e4b31839284eed863889ab094a379311b203e618b7271ca39013c7de07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 72984, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "4f8db90e23364e83b08f6052b9008424ef8de8bc95e13587e1dec390045bf6a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172688, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f6a57ecf0fbcf7f9844c1ded083e3097b6830c04a7fe232fbb2664419851780c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ad1ca189384e140863f738412f846596378beb349e34287f224750dafba03170", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 73368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "fff44d84c9cbe763d10dcb513d17fecadb84cd52978df643611af4e37246eaad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "9983c603032c355a27cb0f623f8e2e330b056e7ce65aa12fb504531ec2e9e7f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 75024, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "59ee0472ec6b83ede2450982c18411f680fd3101054e8c0f3957a00f16090e02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "de1b665209f3be5402728067d86e9e5b6b8380ef4b19e0ccabd8e0af2152f8c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c928e5d9690191f0d1fef091e2c33c57fa767cb6c800dee3f1c8fe262893d0f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 93456, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6165c2d0deccfd19fe9c2f93f26565a1aadbd91234dfdbf6c697a8058e841ef9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 75408, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "4e38376770bff555a6dab12490b4abcb31ffc537412103e3099d675668555c15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 93840, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ce45e81c9ee6ff64746af1be07c23436a93857c129672d922ae7d1029033c9c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 130320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "efc9a9c2b3d0f2971ad5f702447f6ffa83be9e78155585ae9ef0b45dbd04440a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 81176, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "74f1bee4e9bcbca4eba0e6987c0c178829c8f4d4d340eba54704306668310d3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 130704, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "32e8f9665fbfc3c8fc08d42f87ef0260518c499498febebbeb02cd0d3a6b0702", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204048, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "9212d0761e878b06c4dbd5ca37dcc9b260fe05dce2ed60951a67cc9cc3f9475c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 81560, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "9f86ba10f18a47d483485167726217b149a25b5549e8a8eb126d748b53c59cfc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 105752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "8f3a3981358f534695055acc7f9c30723cd3ad9c59ab161327b616918820aee6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204432, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a65ff12445ba72769cf8859b6b0cb6473306f1cae30c78ca12a15603cf4048a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 106136, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "fec0c791f82e9f58a9804490d0f88ea17d0491ba964ae7132a003dfd2d1433a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 75024, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "50e223c004981b11c1427456999808383632719f0b8656924b83604c9eb8c53c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154904, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "eee7615a08def6fea58e8fb5e68847e81dd30893ac93ed8fec0dc1e5e2676728", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 93456, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "6b17574dd78c3aa17a2ffdbc56b8132f5ccb350aa19b4128cb5fdf5dd603afce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 75408, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "4c45a31da201dcf7867e5546b6813ee2254cc15f1182885616cd1d412b3e8a1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 155288, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "cf529c8dee2d2d6742215a7f72861825c8401d5d4edab4f3e103e1202f7066a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 93840, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "45bba55b555f06fa5a76e65bee07bf639e7d7958539ec978b99d9096f7d97d00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 130320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "7b6c72ad7feaf9a9de9ba37fd0dd457424273fed2d3ceba473a976a8e1134403", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 81176, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "366196d3ea6e2e3a699c08f3781aed139391a41175ff4d8a6d37c50f2ea85dc6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 130704, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "75185ebe567c54a8f3908bdfec7b9c66ab9c25bdf31ba6b4a94c1d2f1b70ef44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204048, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5fb35cd7d141562ed275e57085858e7b44ed14756c5a71b0bf1b3b2cee551259", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 105752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e87e8fe96bbde941e31396b11e0f41c114fb9c703e3214b880709ed8728be3fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 81560, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f91d28fa9966d7fe320b79380b780b08b35667827ba9f2ef84a82541860eeee7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 106136, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c3e27f92b88577c568624b41a34d5f1e9b0a6ba75116908c350f6563cc11f78a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204432, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5d12fcd91f2cee770622ef380bc3b141deca90980f8f01e0e3bb69539c5de6ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 154904, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "208764337d965cf7eb705f843723ffaeddf83ab9d17b80ffc9b22aae57f85087", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256u2_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ec2ad204201266a30d963d7617d287634d76edf097e956e32765887e429d8cea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "2425b59c9ccf3f3c224feac5afee7f159ac285c5ee9b29bfb85e7ea72cb31a9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 155288, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "678d12678d79a6160003c2a25f64046753256592f1bc2fc25225071bde363b92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "3fcb608a08aec1dc10f4a1ea7dc32ad5724306fde77484cdb68b9c6207550365", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "ea6a4a3686aa4e6d0ecd1021dcee0a2eeaee09bceb81bbb7fb29a18f41139b79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c1491442fd9fc4359e10b2e5a3d8bd02b25f13f6ce99d8e6fefa69f6595a3d8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "488a32b8df81d6a3f4db83f9ca2d5e746b5f53318dfdd8fc4c3955ab3c81b098", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "52079df045ec52837468766bb590bbd4f7eec37df528c01b9581ba1be035ad5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "7d2d9a8b4df1fe64dbc35da68a042068873d95638de705ce64c781a556178844", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "e0d55cc783fd9156e568fb16abaf846883fda4101152a14ac877dfb84d58c38d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172688, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "a86d52054ba38e442221720c4fcc2c7db463781086c6ee6d5ba17fda9f51d7d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 72984, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "f46c7e5c9ed509d0ce942028baccd538be6383e0de88e7b0ba25ceabcfd8fe0e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 73368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "26f32cd74780bb3ce97fc918364a5f05038ca190f87d3a660493d3061c044888", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "2ff60c652cccad936ffd8bccb498002e8fbe5a1f3a6d121dc57e92e03c409f51", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "237ce01af18ea031fcd1e874299ae974104fcb4bc78b7c5d867894a1132233a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "0c71b8a7f32def431d0158342e579f9d4ca5a69f9096c7ea2a5483d9d71b8797", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "8d189579d6b3a2bd3ca5252bc3039ee25fccdffae48af9ecda34af9bec5ee615", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204056, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256_s2_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "7e79b9c93039ae79314f061acac9221db90016f9f7062749ecc3cea6f979e2e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "78659a057bfd9c0cdd38a7fa1d45abd3e02c5e9f4870aedb9f87a56471dc4e24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 204056, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256u2_s2_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "33ec5350b83f48746846891898e301770e6f297e42fca5b472a3230885938ce6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "23c713ec04e7a561edd4230331287d2ea06fafbfcdca0a78f9623cc72d0be0cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "3f9d6a0f4a7f3163473de4926dc3a52c2b44f021b01e46a49d5fdfb03d0da925", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "4ee489d5b27adfaabb381bb50dfe62a61e2c5b0ab901e6060d93c803293ba1b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "7e4dbfc21280353f0a22eb8e8d1ae0b87c2a1b6102bf69a17c90147a14ed3bdd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "2bea744e1ecc659dce45f7e2e1d4edcb85164be12257f3a7ee5717c7fa481416", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "2201017d8fbc87643c45b705b81b05c4ae988aece73383b2945f4ee2655bd39d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 72984, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "c1f1594c9bb2b8d132a054fe7901ec3b8e00e179e7f6f6581956e98c1bf7d3b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 73368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "eb426f27a7b9d19f140670f334f8a4c7305128e891be46f09a2c5e6b3d8d2514", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 172688, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "aef1cf3a9e963837fc6481ed27edc4385abecf1886393812b6cdd8b5b4d3a1c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "9a1c9615057a93b7787a25fceafb164d7099d5313a5506cc0e6e24439adedadc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "c6b404b955a3c9db2eeffa97663d472d41b95e81fe2a98e3e3508b49a60a693c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "df47fdb20e3331399e6ea5f6880208c95bc24d91cabf936e6dd033e74e5d21b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "5166305454321e9ae763c2db0ca0779e695a70f4559d6363985c44c90c3aff38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "17f8fe6084d5bb5bf4e2f8f8683cf0efc06afe4ec3e9928ca2a6798da32b045d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "4f4a0587a3e7181d95c13ed9cf54b789983062669ffa99b2dd6d4e43707f79b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 512, "40ef9dcb73219ae2edfa1812edb30ae32d079a0d064242725bdc539ac3558f90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "0636ceea4ab0079886a463225a41e871e7ed4f6f7b71d5dbcc6e7bb16f72e53f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "c01da11dfa5a164c06dc4008a59e628d6a289bf981803b18541792287992a773", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256u2_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "241f6950dceb8930bec95c5b01b6087c7ae1e4024550542fac2e1aec13820316", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm100f", 384, "4eeddd213914daab9b3944ea64fb1c0d5839093b4d977884aae106e283ccfcb3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "f0799662b868086b18556dc1c9b4ac4a430a04f11e2b1b3a2929af6af6291477", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "8b9940f87847dd4d2172d9cf9583f10e35d735f3abee4d3e68780ca85b773fb5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "6548decd46d75f31dfbf90fc21e1a833649c580c8c10dfbda38cc42e969da374", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "ae836ed38800fc3f3d662d6bcbcd643150e320170fa6f72357f9179c18f0529c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "253670c7f095837a5dd3ffb6d91d1c20054c8927ba4a2b64e62d1e82cc1a78a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "c1bdb7d28826cad2d6835580edf6df41de535b0f48b92da1ad6360b116959968", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "b2146fee9b839cf7b9bbb64df4ef7ed47088c30fb19840f05b103cf5bb645bf1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "a6fa9ed9cb03e8fac883cbffce592365318c8968108b2be0f302e8b6e02c89ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "d4a71588dd70761ee3420e26fdfb603bc0f244e34733a7757d198bf745faad23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "3592eddb166fdd97879d81ce6879b45df6e0789c68130f735cd940b83658f24a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "8396288040dad096abd318de3159eb0d0e4aa4d7b433d00164c98520f926a57e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "4a52e6c6517be48a8d7033fd1843983f77b250f650e353f3fb302880a883ae23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "11d009f35cb8e3873165ecf983b9b91068768e6d4024381d2740292e5209f726", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "c417c1c2e2b7a7f6aee9a73a113c666c46da56597072351bd24eb7555277d841", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "20934856f70195425717f65424293765245fa273d6f42b8ccdbe3a4a3c7b47d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "9c3256fddb713725a25df345e0a6fafecc412ebd567c4186160007a0eee9d5b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "0b02af968bbff42988c06d3abbcb1efcdbb033b274ab8b844f0f27b9be3b0e0b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "42acb3fd474b792ce2732c89788a0c327705a70c954eb5a90f7e60177b795858", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "3e5d893eae28a2d403e3f563a252de2ff6e5c4f999a1ddffd250b0ef0378c616", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "8eb3aa7958f1484fe1aa179ae9aa9fe7e698dc57dcdc602a4470aa7fd1a9d247", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "e327f573a805178b1a3faf9935bb0e00369f616a7481c2065934dce9006dfa80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "d4476e6f4b9f71a05a6f435a0f5a404be0de47ede5304edcdd7734feeff0b481", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "60fece290090c736ce57b372acb0ca58d5ba814e176efbf946d37ada875b4450", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "0431c3217ca06daea51670ce0455bf596ee4c261fa40ca5eab8bba6f32c53251", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "75e964a46ba1107c24935ef7f29b3ab20f50512c5e4cd07e515b77df9e128d26", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "ae44f5ed15d385825c50fbef17f38108521b005b610f48e69523eab6797faffb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "ba0d56c79d95b26b05081880b14c6a5ca64cc5713f2639433e8301b95cbf9d53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "51caeb15084630769a587cfaf0e55eeb255b4151fc905d67a1def8f76ff9134a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "6138d2d84134e3b7ef98b14f8c0e289b927f0bd5360f851da191d56544c10cbc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "752f7c6b88ac539851c3c0975d00a60ee819c41ed9a1d07532f8bb4a0e853c66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "faaf048e4ccd7a5122a5853fa4daff4e2d46f8d55e2f63adde0d38d7942bd842", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "93f962c5a0d51d9e32acfb3dd591373d14aefdf7596ff8a4867c84ee755bd016", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "af32067346ddcc73760a40661c60a19970334540b72e20cc37b93e788ea42478", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "9c4bf3af628e97eefc7818d0a129faa681bf590d915d77b0c7ef526f41eafcfc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "19af0eafc1530ca3719bbb86a5d6ca903917d9edaafb96193693ff79d3268ad0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "532a634a147398d7839b410ebd76123c0c7af3b3167232058546c94bfceccb3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "ae36242d0d546bff1395383edfc86af0d1f2adc789f16edb4999cd44b32f48ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "246ebbcdef17d54e0f182862b00d4f6b45566d24a0ed63201cc11a8731565a02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "fab33ab361cd43c10cba2f3af6b15d57fca5de7a7a87bcf6b98ad9345426a301", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "f81d12347413e9c91342a5757862e39b9dd7193d31ff4c34580a0d2f84e230f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "31b185c5042a5ba515f109abfb0b63c71a9853f962115f13425f9f3671ff663b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "02672dbfd32c79aea79f817c7be2515d9fbbe41e0fd2ecfc2581673c86d9c719", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "e9d3b3161073d2db647ebf8be575bd0541951dfa12299220014eb964b4ffb844", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "c6433c06a696c5905dc9d546ca5ac0a28244257b602ddbf0e069d74b684b4c06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "917fa25982c76c9aad0c832f861862c9e5a93b89536b1fd06246837e2a705b1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "37500f6abef078b347648e519e6ffc4153877d2661e07f5bbde9108d7f65c001", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "cd7b188fcdcb0e2175a6d408bac7a197e6072db7ebad94370d1b521405aab130", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "737370b1ebb3019e7c3c11235920aaa2322a586e6b92948222aafdc09e585a2f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "77decc0e145774d8848d8d2da1fea639c59c35983d5a4d51c2e3ec9db3debeb8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "d85ada57108746ff37a6a8e3070ecf4b171438da113670ed97801d1c43d53f0f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "3a2365f9024c8e91a055964996c6b732072939afd2edf5a3feebdd6a5393978c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "08d6f7d957698b617855ef4c2f283f0801b0410cf3fa02932fe9b1ab4ce936c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "885756ec2c42d2f6d4f83453a842d475b7751e94438bd75d2c43b37973a7f1bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "61e7f73026178effe48b6acfda9b544ce0eb59ee2a213196f0af40010e798bb3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "397d8b319441525b4b7a17afaa463f2b95ee8645bcc28d044a2d0fd5b0707d05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "15810368b9409d4586be7cda2a8a6708a2a2edab7deec0fd830b0aba6732cc5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "e6c6d387efc50ad58748ebc776b6defe6b2a5d7fcfa2af0969cc55c05832d2e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "49af65ba172eae4f185b022cabaaae83f432ca2242ad3f40279b8e75fb052e7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184592, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "be4e8230f00ee0e6b680671ed7d5ce2a96529ed862b1df12a1f5bbf511ff2984", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "17420089247f3da8f0d260cefeb786bf40c1fa3a87950fbb77ea5b934aade854", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "15dbe708246dab165af06640bba1c4e8ce594c6b29aa92cd38c3c9c019ef3002", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "51f2bc4f4688922fe0c7faa81506e1d7bfcc13f19c08bc86d7d5b21e701e05ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114960, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "50967133c850f48c3b33718e97103eb6aaf7e6638b668acd4a762e26c526810f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149904, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "050c77cff3ceb0753cbadd730ceb580651074c61e36d74b259811a1e171c2f4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115088, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "6d960c666a8e03853b39cac175c20b72e75b95ae70a588c7770c061cac06ef08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "f58476695b60bcb039477ab2cd9ef02be6c4eabfe97fe6cb8f20666d69a99adb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164112, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "b6d019ac9324fd325d04ad228218ccea5f34fc5aedf62013c6a9b821735dfdab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 201360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "cc769fe575eaf2ebf4da82b73ac6f5000586aec39f38178de9bce86d7d510606", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 127248, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "7d2d26ee7902ee7b6275251a32fb4c072a194fa3696439541f880373c745c43b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164240, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "d6e5a65aa6b5c87e82d03b6d88590de227f937ed1444b9356deba61f957b7a42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 127376, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "2812cf5452b6e6cbd933f177a22d8649e835efeb2098a2b5cdd57df6589a5077", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "92678ef07681716a061c5767764db8689b6fa716ac8c6ffc194fd8983f114bb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184592, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "4e06b276fbdeeb5082600600b3a131615a633831992c089a0d633b5d217980e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "2a2a303b8f835e99eaab18572bc1e67573d6b9faea8596901ef6b9faef26cf58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151824, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "fd1390d24f539c074a1705349a259e1dc1c434a2fc6203d27e790f4474381a74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "c298284a8a2130fac9d0f6f795c1e2208d8dc90268113434a3b5b413c47eb6df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192912, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "a9617143e5bbccaa9eb2514a3e86a3cf5f81093642b80c2ad2c60de52f075556", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151952, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "2876c20db601f394321812f1ac38fab9539eca3c95b3cc555f57c5b63f2e8716", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114960, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "5fab682f792b7d5e5a400d9b95e9becb58fddcad24328b222274903d7c252632", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149904, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "e9eaf69a99d7e70839726dd74fbc4b5fd0f2a486459d610b45c730e98af60f98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "0aabff9c349e831dcf11f0fc1640c04e13daa934ac310713f1c35fd8ae39152e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115088, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "4654282191f8c0c2a84696a5f46ffb0bdee2d1f0491479dd00eca9739646d353", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164112, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "25f968f183a0eed158be342ca6a75f17e7f24f121960943f8b92541140077d2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 201360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "332f438f4da015f039dd5830aa818944d4715619909a6f9354d23d50a0af2133", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 127248, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "b8459584e580156752823ea23ee4916b96d89797a2eeb3d6516f38a4aa53f7dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164240, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "25ba3c33877e2a949cb4732e5704bbee153b54daebefc86b639add84c4fd8143", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 127376, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "51c548d87b0d946285d5da63ceedcec9d3eb341c6394cff6e2895d7296b11366", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "bcfc50febe1735c81c9b5d022a2a34094697b765cf6ffe4a2574cd4a1c55c7a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184592, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "54c49fd1ed6c16bcb27bc3ec2b95f1107e01702549e22d7a7bebd9eb773d9117", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151824, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 256, "e76fb4e5f6211b3c858691e45e2f3374feecfc8b7c011f840dfdf9f9dce2cd84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "e13e079af27390aec44f91b835931ae2a4b96028e8a3631ded411cadb25e8418", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "c97ed28a86cec98ab505af1d921baf86f062d3bd7db5753d116d2a895b0b9019", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192912, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "de4b2e372b79c26aa4fa23d3c05d8412f3515684cdb2e9b48a46b4fc8660ca77", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151952, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "c25ff7ea239f678ac4d1709b52829b59b0b58ca92fd23316b702c49255b79f06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149904, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "4bac6bc21011c9bedf5f87ca7471b10bb9d0522f6d3c9e3db7a5886ee8022701", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114960, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "b0728b4a9c1c36c46aeb2e7ddbebc224a7248d7a767beb016ae0c204bc11785e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "3872ec2044d24310703343f534bb6d01c542c7dd3bee35d3dd0989d160a1d44d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115088, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "925e0cc4fa6db2826c7118a2a152621f0d586a529d652cc428648261f9f1ca0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 201360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "ffb9c385f652c6a5315a95e9bd7ce0275d0ab27f0f0b612e3564d485e876df53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164112, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "7e303c8865f2d5ba82e26171e788fa4b6eaac5df981e82c838b360ea7f99fa0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164240, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "3acf8b74bc838e1cc6af05c16a33ed2d778a74761493c95857b3a808d9e3c13a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 127248, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "8182e366cfd6c9ba55c4758b2c39f746e9e2930ea7e94a90181ea5b50c0fb761", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 127376, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "c784e2a2f403a32411d6edadb00ba18710fc19d4e9d15aeb27e4192fc4eb1c5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "69d2366bc8f85604666eca21b71b32b0eec052a79e4d293cc0003e35df266f61", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184592, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "e01ab4a70138cf9de3a2343671688955ab7309f9ca5c541203a3490de47e7f90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151824, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "db3332a6abd412bb1a9200d077c4f2b86bb972288c3acc09da490ecbad107797", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 184976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "544a883bb38f204139cf183f6b51d4ea1ec92b99e8d6abeb1968153e8a879b92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "02fdde252f3f631e478377bad747c6f383dd62d1dc32af7bbb3074a47fd298ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192912, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "1eb46faa344c4dbdcfc502455106d054ad49960f55d5353fee7421833d0b74f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151952, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "05e1361ecf1a691c8856cf1e5b99f8bd9c1397d4215911d5bf8d7c6b73074a88", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 114960, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "6d2cd445b579c2e4e293b7597f5e5f2a0d3f63e542ad157adec37f338f5cf30e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 149904, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "2d59b6971701c77676398dca1074cec4338383a3b2bbe403062c1e81b3d4d74e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 200976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "74b8e5e7f3af074706c8bc84ffcdb76b17864515573db9abbbe4719183e4c1ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 115088, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "9d81b65c9c3b25f62e96d011d170b5d6a4e6cfc0bca9275233c91e6ab1ced8f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164112, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "80afb04cf019adf3c8112d254a9b09eafeb7ea873303b6de66ddbaf9d96a690d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 201360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "db521fd240e3a9dcfb720c151982fb9d5a99db5230e5409935102253a55362b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 127248, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "838f316e3e060fc3db56cfc04370f80e6604c02f5d5f5b9a0099c500daaed218", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 164240, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "56bcc2d8cce021770dac1c235e86b21f971e45b92d706279b3db9fa47af855f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "8a8b713ce235ccab1acd788f7d5aa3a59fb8d01685c561ef2893d7cf9cedc85b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 127376, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "f9f6943bae2ac48d6ee5c1c7b3031ba71129cec9bbccbd2443ce2ad4b59a9cd9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151824, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 256, "9c7c90dfdafac94730c39de6fb05adafdd0a7df8c92dde86003a350bba1e4cca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192912, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "28b65b47f6bdbeb28ce88ed9af8b3b147a67c78fb6dc3c98472f3bbd4cfaa0df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128u2_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "dd32f15b1ee38400f2d2472c43c50865c0893ddf07d4cc92adf0f0b34903e244", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 151952, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm100f", 384, "ad53963122fa41695d24556273eec4da3d519660e1b0e4d1ac4120a176b03a27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "9788a18b1dff62dc5f68f7c4200697a61ac25c34db2d68141ec68b90d56ec763", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186008, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s6_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "635a0d3735ea94d12d373a398336d77233b27d8b6a23144afe2a8efe81c6a33f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128u2_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "874d751ec67b7412d9cbb130f548a4fc43e77fbd0ca518aca11fd363b80fdf95", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186008, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s6_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "a5e3a37fa5e13f8fc2c07764ce4b3980f617504be9ff95402cf85e26f31ca153", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 64
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 64
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "231ab8c58ee4039bb8bbef3fc83e334988d4ed30e404d0f345c91079c92652f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186008, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s6_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "505bb96609f183d0f984aec30962b2b6f27e61dea4de3d1207652a9b0a6fb7f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 186008, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s6_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "571b39c7ca5065ed50238d06c0c5192be04968d3a169f8e2caaf635f39c7027d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 64
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 64
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128u2_s4_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "ed93766ead4a35e75d0935849cbdb6d74d1b691094cdbff5082c8acdfe59e3e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128_s4_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "af7b79efe0b986f9aaffee0d07d993e4a7a177ae209eda24c88dc7434669d945", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "a04e203e073ee2cb554cc66d4778e9d0024b1dbdb65b8e094bc9886d47165f73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "10618cbf1d1ca964a7cee1e19250ea6204b3961ebd34b7aa0f874f74b8f49c60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "fc699b353b4fc73d986118e9e65f7f08cfffb0d7106f2c191eae8936ab69cab0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128u2_s4_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "53b5d1ecd23fe897774303a8465b6a278f2fa47dd698e9539732a44d29cecded", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "a406bd1af5bfd301c6cd479d0aa232f5afeb1a786a08f20edf1ff5bb04f29976", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "eb01d84ddb53044b0fb231e1b59f1fc91503a861390b62f801b0c9b18410607e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128_s4_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "106e196cd323f9dab0e9683282500e948c692e600cde697d5c08ab1bd4a037d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226968, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "fe39e6a459f69b7f7a935064abfabbcc6d90465ff1dcb9229ccff674c5717634", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "370ab44c23894b253037fa8be17b8dcf8c58013638e7310c64d8d2bf0e27fc59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "f5d3ad925016acde765ad17150838e7412fa7e5d5b6125c018830deea1760cbe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226968, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "8ac9d2e53bbbfe0b3ce115876b7584908e4a052382ad606f607eb1cd1c1ad6ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226968, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm100f", 384, "ac1af403822fcc3810accb2a05f12ca26275c9bc2a59c9dca4f7d04e5e4173d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 64
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 64
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 226968, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 384, "779972c81e7e9eff741e0f562a4d438c53bf2c3e5ca7e24d54022a9f090ae75c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 64
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 64
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "c7fb3eae08ee0b2909f76bf3b4608437c2c4f2ce6dc9ecb5cdfa8e83d95a33a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "ac86df757fa78f6881046bef59b89e025b191783265126e774b95fe381e8b53b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "db3dcfd67cef6a5839f711fb9a2175abf6e6435e86d4e4e533ae83017d6661d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "d342f53cc45197374c2ee2b78d1535609035d28316e5217de46477bf06587f57", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "280bdc684a47eda9f9779f9e9e2cc654ed30fd420a592dcc471926d8f896a108", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "1b7bbcf184a345a77460e7bc1b287967ed4399768e2a702482bee8b39568e94f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "80ac2ec6dcce3a7e30efc760f72c30166fdae95f04894dfebcbd3418c7adf3a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "0379d64c0c23f6a4e6a03d99f3fe25d4624864f2066b0240ebe2dbb051e6854d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "475e454474f5c8f77ade4bf96a2e4e106698be0d4edbf6d487d8d9e58f00329b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "d1955d7582dce0030b911c757702932f2faa3184f39dc97a8e33e2429d0a498e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "da1caef1353b39993da55d9090106a9b9cfc0427ed289c206475f296d963748e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "c99bfb47c9152b42ec30e4c4993fc5a25f49725350baf0ff689998c88781769a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "926b1dad62ca5e9ce6a7488da4c0e6c271d41d33431a21008da12663bdc415e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "186cc12fb91bd5d267753137c912e2b00f637df8573751de66b8988083f34ab4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "cd8e8b7366bb2a83e9f4d793f5eb9834757d91ba0d139b4b575e5e3f67f1dd80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "249e140a76a2f0ba9ae1184d4652332325724d24b46d6f1a71165172262fbfa4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "9fbb1aa9703aeafd9f1818fcf8749c2cd0322876d0234c0141130860004ac73d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "bcfc85c191b2ef750c6af9aa27cc7aaa6df8f079d380b01bd0b1abba8a8262b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "d55b622070c08ffcb16d49697894af283dcc601c9d62d07a4c6765319a88f6c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "a25c66bea5858cf0c12740c86c025682153f63c0998878574e50320393f4b210", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "4fe40fb0e287eb7e5c82ec54229f2a63ac3338b96ae3c89602bbed0b62f710c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "217fc4e2f421c826f85e2963a7b249de753d0a71bdcaef2e45b93e729d73bb9d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "1e8d178b5305afe41f02ee704165edfa5133b4889209f2268cb95eca5998e354", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 256, "009f3e47bf83f3fead70f2ebe278aecc8ec9f2fecb26982d661a6a47332ec10b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "1a528d75b700c6eb5d74dfcd2e3d66f37c8eb0a48bb7069026999c117e8b9b4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128u2_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "da775f122fa23dc2ae2770b85f43cb69c93e6daffe5fb4c3d7cac9f65040af32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s8_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "06de4d50ee85e6f2cca825eacad2d8f9b16d2824390d21655f5fb932759969e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "2a23aa9c71350312f0622ff72293c47412129fdd2e04d3a175c735b74d3615a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 218776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s8_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 256, "f9dda6642c8a1546fe18c63f4a990d2a06a467efc8f8c2083a04cfda7d9e0eed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 64
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 64
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 74000, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "521869669334f02eeaba77bf11e4f8f949113d51f7e8819ab723e0d38e655139", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 88336, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "296342a14048acb1820f3bc29600f105d2125d674b58a0144b2be751fbbfe591", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 74384, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "932ada9249042bc8283d1e0c90c7caea4b984e115a019bd77737596f3c53b1e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 76056, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "1ec19d220456a4feaf2dd2f6c5e70d17d4d53d75aae2ed95030d4d46a2b73ae0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 88720, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "42377e5a5aac70ccab7d77fab04c2d05a2f2734c1305d5342bf805442791be0f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117008, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5eb7a226ef26eec1cd51e031dc7eecda4f08466a1d1f6345c34e8bd2aa3c6f6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 76440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e736a248f0cc7e930cff016cb40105f0564dbd97b7c582fb31fa100de58c5d1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 92440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5bdbe4274210a4d1caf11e52e77bacc6c95abd0976b4d6c95bdf865beb4c398b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "70bcb2c8644e494fa977767f6ac2e7f1cac47829d0491a2d085613948d4f3584", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 174352, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "8c1d21a7699ce005211347d88d5455c8c7723f60f5df029d59528d25ca223461", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 92824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "b0d7bf64351b2efc95986d1b5e4acf5fe443398fcb1046399bacb46c58401608", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 74000, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "181e1f00bcee8be6cb533efbfa62cc39fba3daaeb643b4a96b659bf4d24835b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "71c063ffd494f3936ad1a95f1f1ddc3ba0ec7830c14e04949135f53a5e1707be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 174736, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0b228831928b8e7f65331521a69f968ccb54019665e20243b0c2b870fe7c1687", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 88336, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "2fd0a6a6d9eb33918fe5f0a89c38bc6c881dac5d2255c53d101e9d5aa58e17fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 74384, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "037a7c942a97a68523370ed8b59fa6c937490b8c441a9d71d479b7f97a4a3b57", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125592, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "57c0ab1c2796306c3636ec25a2b9b1f6023115289769319acdbd5285a1a8bc3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 76056, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "1c13128a6723385fd9a99ee4ffcef93ffbd05803868e4c8f321b830f04174877", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 88720, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "49a84c9ddaad809582036568575b5890e31cd649dc0e0ddf63da00ed0882170d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 76440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "5eebdc8ea9cb5c219eca29893ad1dca98c6e0f96f0aa1ead750ebd5d14fe0819", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117008, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "26eb594a1cc82ced9b8bf971b34cedea4086ec01655319a0a551e36bcf324111", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 92440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c588ab6601c4e7c988c24fa819ab17af7c0c8bd58797609cd2914cf905c47ca7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c6b81bec497129b5384e24bfbacee8b11ccc9495f1a506bb83d8a3fbef9fd4a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 174352, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a560cad3d4709464b6790921ac87eb27807ed542b678d4aebde4aeac3559943e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 92824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ef35ff6c1e69ab9cf0d03d0b33c3cc2afbc26637b9e236de7f4a607fda1b8cd7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ba0c80de93db7ed0e58650df44b01e2672a5684e3c3f86b04c5e832e7f99093c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 174736, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d2fa3af5d41f831b095743a27758d0297df090e832fcf70d30a8bc24a34fa698", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125592, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "92320e45a6d44261119b40bb8029acfd8843fffc3a81887602a88b017e849237", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 78096, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "fa9d0ee0bedbfdf91a7fda3a5faba8f4dbf4f42b1b5d4333401dc1f527e3d5f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 96528, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "0869ce8371d2bb13957906012bdc33198780fb1748d27966363f20c92bb8900c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 78480, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f2389a0028a565a6845cffe512864bc258efa6347cfb4bc6b97a39a970ebe0c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 96912, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "26db0422b1b17ac71886009b66b42a5886709bb6b4a8011ef570b2760c43676c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 84248, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "955f5ac66d0c0f967a122420f0d3985f3a1ab64f8ebed3c1176d5c225b2d0c1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "b80f5c097cd8a58c6df39bd41a3a21a2b2f1801895435915447a5f13c54bd3a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 84632, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "90c09c9142dd7b0d8b94790642e37a4f484cd8ea0c4d1c8fb25379b883ca3729", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 108824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "cf92ba9c0a09b9d7e897ecceeb6c4c3ae49957a7755e7f4f30fe4d4f82911afc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133776, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d986594b46963e2dc202adafb407a3678f0013593fba9206543af633ce88cb14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 109208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "2fd684f2a94871b1077def2d96667997c859dfc96009f8ca0928a5beeb2cc348", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207120, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "9e3959a1ad6870766d98082bf912ef44e6bf7cb09956c8b7246be43130a17e37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 78096, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "fc0645380a0019921f01b070cab66726b30b9d84205ee798a197be27ba718c62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 96528, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "b5447286bfa9ebacc05b3ee14c4f153f4512ddc3f45648b00c1e63d75b40c17a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157976, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "ed8216fb9c7f8002415c55f9a9ca5d74c0d54e8289aa7ce6d93d67f10724373a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 78480, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "391f9bb42f6ee100c9c581003c11fac2389b3f3ec798fd23ca5bb28fae23e583", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 84248, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "696fc29f50c521f740b17d142882fa82d1f8d0e95c6dc47e772954440673b9f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207504, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "d15747108ac05e039077349b75c6c08dd2dafac977448174485a1a5e37d21903", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 96912, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "b811cf111558833199efb3bc4252b5023d15b7ce7f414e236e99c6012e2e5c85", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 84632, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "16b8671fbb9adae8609f80e6a06d9d4567893e10dd0e349981b0c74ca282cbab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a48d1500328fad860928b8cf0c72efada0143f3f7e6e0c4180bde969dd24a718", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 158360, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "65edd399836177ad6ec5a06310e6b657ef095a433b0f0589cab803e01ed8ebec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 108824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "838a71a31bc90be04e013739cad29fffbce55de2361c1bb161abfb415f3c482a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 133776, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "a7c134810f3e83d43a032643892cec6dae72e007f230f97ac68b8169eb8219ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 109208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "e7877d6b37687ddf55b5bd3b52ea42df918403e9101b092e11fe1db589ad7936", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207120, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "749ad96b3336759ad5e7414031bc8ccd69a1ca619fd3555f1333ece80ae69022", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 124440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "f9160e6c504dbb5f324909c238ae0b9b383e32a72d720c8284a6b185d3818891", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 157976, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "2e9d7ee4567429b51905b6a87dc639ddc96181d932703ed9a9516b09b812e350", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 124440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "de97abaaab8ae7a343c9d9d1958dc0077bbfb0a78ee0382d656e189e35953da8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 124568, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "c31b69bbdaa69bcc3e85abbb41e3ceadacf9d6192553152fcdc0ff387a5f0f44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 207504, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "cefb9cf0a837597f6806baf16484ecb688a754045aea11dbd8d38ac179dd6d9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 158360, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "20f6520febb6397c5d1be2b6ee5ae00fffa71f6d453199d8d05954aed603be7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 124568, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "89de5f282ca74371a3973d23e7394a53b10ece185dcbd409de99a93c17902cfc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 74000, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "5752bba02fc6d29bd121b328403c974b08a835987f61c46062755002be1e3ccf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 88336, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "6fcd608231b57ba79116e7af05080fb86d6dae4ae2da32a8057447458c9d87ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 74384, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "2bc6d003943fd10c2fb9fc45aac00446e0d2e25cc795d48856400fae666a160d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 88720, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "7dc276ef93e0b4cfcf0f426fcec876d89881aa1f99e9aa720245873ba86ae3af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 76056, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "19c93c492705e449eb130bd24d92824c42b7b2e715218d824a97d1fc5401c152", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 76440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "8c1610c8dab9021bff0edaf0385684251f02d35f1311fc3eeb18e366213309d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189976, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "3eb018175ce77d0d126b9a8a2772e51a199b0e21c4a27b7adb6aac491f07ec20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117008, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "ced0a94da31b9968cb4108c04a62d2bef707a8b4d9ec037819060b9c2a67a76a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190104, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "bc94170dfd5bfb804216fde03dbdda89ad342ba627c8b05a5cfa656aa9dfc780", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 189976, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "227b12db1dfed5b9003ab5bfd69b8f41f6719792d6f4155953ffb2b3db0bef5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "09ed678368c29dae07b1d6a802b4895ff88bd9283ffdf391ba2f7518609c02d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 190104, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm100f", 512, "31de3a160bbfc3240ed2ceec5b451b61887d911587aa461399d8bfbffcac0c1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 92440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "b1f615aa346aa9bb8bfe56583cc280ac3f2033679de60fe267c8b9092faf672a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 174352, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "e73b53f59cc164b3fe5e64b0e6112f81e3d082dce02c75ce74df8afc84c46fc4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "fb947f9f1bb317671e2230545c2c9ac9c72c1a78dbac60b33c630cd4ff5e4c67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 92824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "ffc80dd883a021af185d43caafe29a670ea7abb43a90234c2249dadf38409d5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 174736, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "688c3dd9e382cc191306c0eb3500ee90d951689ee0514d42e2d9f4f191a02e01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 74000, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "5fa6e7fda7d57d72edb02a03157a5cba738187bc83ccc13e4bb72094ab2f59b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125592, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "0ead610bb1ef6e7dcdb27c05d4dd508f42a2cc1bb720b2b6a0cedf2546263856", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 88336, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "4684fe291f40255ec87eb05905b23d9f60c800b1a15c45121d99aa10dc9513b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 74384, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "4b33d02cf6153815af80673726db6b43941d28ce2a849ba5df921df6f6f9bb79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 76056, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "d29294cd866a16916a2bbb677c199b4c17831514ae5b37dfaacc962a82dbb5bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 88720, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "0dec63604bfd7094f18dea437822a6b1346a64049c3448564c3c8a6a4d2a039e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 76440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "37acb5564d301c36e76e42dca754061378a0c74ae4a6183dd4e6ddda714caf49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117008, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "aec1b848e021f4c348ff3b640e2f273941ad11b01ed3b7a7b569f07354123061", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 92440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "048b509b6a10b45cce18663cb29f3a776378c4fa30a579081239dcfdc8ed518d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 117392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "025bb8f64f8074ed2e293c7a1536311fbc06a7dc89d447ce0b9083f4e626da91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 174352, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "cb47f69785afb759c7d6b267a7833dce8ead6906592e247cecb4c10437c702f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 92824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "fc44b19ba86dc0102c2eafeb9acc195ec9f0c6b517d3bb2d56798dc0834e4363", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 174736, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "d0ed4b89f88f12cb286e0d48d9a7ef1de0cc9ef615a55f2cd59ed4c19fc0d0a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "b619596584ce39fde7a4df630965793741c0ea75848684bb57f30194fe19e4d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 125592, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "cffa764fad050b0295766b09aa8e9fb8e424d33afd92206c13a128d1f4c6f38c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 124440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "f2469b3584051d1a7f64a587dbdd97d70634381acdf8dda2c09e8cb8594ef08b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 124440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm100f", 384, "c8c49b38767af288a3036b3e8af0734a1185961d049cd9c67c6573e1db796e2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 124568, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "6b82a3756147651935fda2b859e693c8f6be6768b6bceadd71e75db4183e5c82", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 124568, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm100f", 512, "c968c1c623e90520028500311256d137abdecab52baa54d026d7e48291ece6ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm100f},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "899e0c36db8f74565ed48efd85daceef3db05f9eaa0e984dca8bb671074321be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "dbc083543f0077c2707c8a289f7d335afbfdd874b96ae8ee80dd31aced882aa2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "b28a95be03f1160d9f103f95a34b4c2c95e354d014e45bf9c33fd8034a9b770f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 72984, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "626459978623a1c519c2096470f5940b35884655a4c1889b8fbc55cbfa8ab0a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 75024, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "77002671578e42e38e163ec37a5c31c75b51652f840d07923e74ec130f974e98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "d0c1545f49a88abb516ac098ec5d458c90fc908e217168ec9bafc8ff8f15c5fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "40a971f2d924718fa68c181ee4733d28274222468e7a7adac3baf8b9018bbcb4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "0c126236eae736f6b9b4b370a499e46aa19d1b378cc66dea15fd2e1eea626bdf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 93456, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "92542586d7c79da73810d9eb6e8b5615e8c0c7b54ab201625c38efe3d7e474d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 72984, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "9f840b16dc24651b525404cd8e809db712cd7cc21336eace4298820457c1c502", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "4e10d6f0782bd20a4c9b438729d13d830f154c4822c0d4634ab1ebc9c75b2871", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "6b13617a65adc67a9ffdb2af355eb26a5f61edad62bada5babf499d1a76aada7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "7bc3eb9000a443c3fce642fd2a6a22ce5fa413f0c5ea3bc6ece9c29cc656b92c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "81fd8fb9035d075ff7d4dbacf5ffcab8b3182c7b1e295cbe9e881a813e5db1d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 114320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "d45b5782ada6cdb3ef52232bb50495eac34440d4a03adcfaa6ce467637948066", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "3a163d0f067955a62c7621d930d42e2a258f7dfb28c73f7a3c779e13270abd30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "10bd96af44ec4f2a9d008663f8aea4324d289ba598e08a633b3c6e4930727dc8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 75408, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "a17fa28255cd069ec93ba98636bf72974b362c699d18a0bcc13199f2123d1854", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "6205ff0e312e969c03ddb28c2c7893414ca51cf87e73e9f434a726a5a3da1b48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 114320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "2965dc67edf640644faf7819b838161a7646ec0eab06e3d88d8e89faa0e796d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 73368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "36a295a34f7e1c6506dfbb105a91b477b72541f70dc091b8e01aa85fcab9456d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "6975fc9d77baf1ec490c5012f8ae35b4069b1c3097f734de8f1f59df98350f0a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "32d144bddc9a1255cb07fa8c3f9f3cb13b2834bfdc547b26b04029af113f6d80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "ed703e4ab924f7f10a0d0579f6b0f48118b998e07d054a4ffbead1bee298c294", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 73368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "2a54a573eea3c5c95e75c482ebaefba7304bdb755e1ff972b7b4c2a3eb367be5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "ef8a6866e531d94d92496ff5864cc1b7ab183a423b9b608996fbd943e6b3c987", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 93840, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "c61e78debadaa29882813ccd03011167ec8253c16a03fcdddef4833548fcbdd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 172688, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "771d6dd9192ff0136da1b76855033d4c441e7945f1bc1035f8d21504a91f5d2a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "731321547a83d87a42ece9558e91e8d5a24b221e4cd3d7fab637e5693582da7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 172688, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "78d65aea35914679b51cf4664e77cb724bb5c5a12a50010d7928cc7c76da321f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "0a44c8656d80a41a40cc4fd0f97f700b6a955773e21af8a9a2e7eba1f39f041b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "0f4cfa69187884f69dad5028fecc23ba4665c1c36ad944c6b729174209fcf8c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 75024, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "87c74816c7950e5b0fd3559dfbbed26e7c80592083c419e44d8827951ef6e083", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 81176, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "49e6c87e115eae018be45fdf6e00ac8bfd315c0dbb5ea86a3fcdb9785e500081", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 130320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "dd936301560b7a93fbd90a7f4cf144d06c6071b7809f15bfadc9c7be60af514e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 93456, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "f0e48b3a2b4c6d65123764dff1b654210e3df0bef3a4648cc984cf754a00f968", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 81176, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "18ef0d91eab35c499413f2179453a77ae8d8ae5261aafd352887e9d06a96d76d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "e4d7e8832bec2ad33fe066be7d6d7ceff56b6d651cf84592fb0234aaa9922a42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 75408, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "473f6f00e5916f7ad2044d4124f67327cbe74a7da6f53bdce6d806dd9f2fcc43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 130320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "a6b8935e5e040bcfb4b90fd12f0add8f160685dd4a2def013e056e600d3c6546", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "10c02a00ab7da4ca8eeecede328fe7e70fbe0f475836ea767892858ad3a0195e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 93840, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "9bdd68ff8f12a862a8c8d233d93ebaf4293839e55edd5e214930fcd85610ec51", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 105752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "ffc48d61136e46918d7cf3f89a56c40295da98f6e6657b361aee83cbfdb1f71a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 130704, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "bbf673f061d403ac480afa7fc1ea25005dbad80b11fdb794685be83c5282fb11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 81560, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "ba8177e6906bab2adcbb5d1ee673dce5280a81afee27cfefc2e7e417b448478e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 105752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "59d670cf7ed9b20a457d0ab47abed4866e6aedcc5cc705d736ea9951867738fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "1c7a82a6e3e3e7e7fa9f7946586a66eb4006eef3fc594a98e4a789dec70c5f1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 204048, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "45f42a8213d1b11af2f1106dee6f645a5a917ef6f369546b1cb93e17aaece831", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 81560, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "5631f79f571bfddb91c0a706f1a497bb1e8a72c43f306896f5285fcdd648574c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "dfc020e177109c611a3915b429841c6c5af54c0df8b885e6da5f6d741ebd7129", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 106136, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "6b12a72818041b9a8f50c5767277424f1011540e01282a4c1e0a82da88300f7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 130704, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "dd151c07a47f80ddf51c8f1df69c6f4e86052bc28ea1dced1041dc5d32aa0642", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 204048, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "1e1a338bec98341557a7288073b376403bbb273405464566e6287d8fc86a8c04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 154904, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "7d0652b38ae51e2eb3b15a9479dd7175d5b0092e44cf3a19b93bd1cd7ac7dc29", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 106136, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "958261cd0876f87d8a64af18365ddf674ccd100ddb40e35b70f64597025be484", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 204432, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "aeb67854f560e1368bfaf04ae0964dc0e271ca3f7c42a94eccfa4354cfcc46b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "98deabd92a66869116b796e42fe7ebe4a23ad9dfaf88562feab3837c5940ad66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256u2_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "7ac76755edb8e5207e410cb689a14e53a096127e6f3707ef812a2f740e69f54f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 154904, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "41abe74d9a6a3432ba36043e888b7ca2a112839c760aee0cbdc8b441bf346ca6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 204432, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "67d3ff931943875befaf0cbdf43e44580216ae96526c369e44186a78aca0f2dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 155288, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "2033d6bb84cb477508ec6ffad98fff34367d1d307a4913bcc72a2817924bef49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 155288, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "90a16e40957c8f86387915ba4347d064b646ae94c3ba38204c47b1f664354047", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 204056, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256u2_s2_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "0bd03a98e9893eb8a02a49bfa037353202c02b645587f70ee2e87010890cd3b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 204056, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256_s2_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm103a", 512, "8189a7588c43339c5a7a2a949f4297dd4f10a6d500e3648c40ae5a76abaaea41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "f4b8c6c78c449151972b712295986cb1e5a3f004100c7ef184574c0296a83c64", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "a19ff52fe0c8a03f9756efd560d8dae6bd9102e22005198e9a45959f663b554b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 114320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "233b7f9bb0fe1c36a668c1adf93eec38b3d58e1a93246748b8532df2146830c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 72984, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "2f14511ee2f28f109d34059cc6eaab317213d405e3830ea66143195a06079413", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "da4bd544724af7979d2468785578a336d2cba05a90dc9cc7764e6d00847f336c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "cc16f622594572cc3d05b42480f52f763d50d05561ffa9d76db6a3d47d83dce0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 172688, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "d72218515c3ca9e619f5e238a0b3a313a23ff198137e453e4d36386a0f7170ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "98f32a1601dc192715f4629ca3bff26a50dd7490aa4a14d01809c79c3f604fa7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "1545c52a50dd74c8d7a33021c676b95a7b4ff8528eeba3c7f3e5bce0a962d6b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 73368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "5d372cb2d3579f3954b8e39dd7f9477e90aecfc136ce5be6ecec5d5800e666d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "3653db9209e3523c3a2239e393ccba173bb679bc2e503860e7dd741ff62c5843", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "932fa70e3abc8b925acfdb9be12a3ce885abb3a08637c147982862e808f65bee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "9cd308d552b03245ea951f7e7b29d5bfbb0aa33f07b8401e5aa6cd0daed341be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "3c31bed548c2ca1f0e62b50428f3fd76e0f16abe827b7f5c0f7f4b9cd8360bd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 72984, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "7d4ad4b7a3b3330660076448b71e673510fa1a4a6576c261dffb7ea70441e191", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "0b13a82478936697185f6dbed2855372695cb7e1ae9c064ce0ac85416568827e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 114320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "be4f3982f3be36aa694bebb082b3dc4f1bc2878cce4c0a3b4525e01fc6008cbb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "4cf8502cc83e006735a721bdacd27d98e3ba171eb6f05a550d4e7b1dd2082ec0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "b5ba2f10168c2a58bbee80fe6fdbcb805862bf616a2ad82c591764f2f5fc6341", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 73368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "b9f84f091a812254c89dce10828f090ee1de9379b916d1b02901101b2f088de9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 172688, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "0223200a72c7c4abda2a89a8ccfc38eecfee2661e587ea29140700f557145b17", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "f95e1a2996c8baa0d48530cfa135c4933f2c46d8fa3d432cd8085f09f7d8614b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "b56845e39fdcbd2583cb4495851334d0570e6ab84345c5338bf8019295938478", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256u2_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "95360a36e2c3f5c622feebe4727522f1a152e9bc1a03f26f65e40a9c2d09f9f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 384, "f95b3b9bc17726ea461949e439e51d9881703d7ea23fda402ee33b828da5bb04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm103a", 512, "6d270c998933694206ecb6734cc60a6184660e4d6d68266d2d04ff549c8c1e1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm103a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm107a", 256, "b5e2e26b85c56855eaef6cb93005a3d0e433272f3de34ecc8485916d7b292806", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm107a", 256, "7e72fcc37f4c5484cee825d10313842da2ab8003abd42d0b9ff53d3fb4100b18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm107a", 256, "5125dbdbc920f4f7148eae8a06d32a5cc3cf57dc9c834f2ba7a6f6548ed66020", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "efe6028732b295972e45223db310c934a1b38522c1951db1209442b60b76c509", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162320, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm107a", 512, "5c135ba64b317aeded6a4796f87bdc2a61ed9a1c0b761f82cf5a0acab720619b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm107a", 256, "6df04cac9d1f0a0f975b63d7c5d5c69913499495adb3b39c8f33bafedb1fc2d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "f06d9b2a8fe54de7451acb595c5b4b8d1e0d63eed87aa1c35c6c6ffb3a82209a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "e25426727a65ba2b54e2ff5034db1037ad1f7bb847bc133257093b1dc05b9c67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm107a", 256, "3ea8a6a875be192a576a23f893909db3e50c9d545fe2942909ff03b8d9c3ab49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163344, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512u2_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm107a", 512, "0b6b577b6ac6633cf56331ce9d7011c8a5a50a88f109edde2eff09b510b65ef2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163344, "bmm_Fp16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm107a", 512, "06b8f396c2022c8cc89589267773d049684d1be44f1f313d5e51a5b4a340f6ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm107a", 256, "d293757da76142a84b74359e2f0cfb13577597ffb0456336b17844bde326494a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm107a", 256, "52652b9c07cdfc6ad0f1294050e761e55f952a8930162a6ffb198044b99b1233", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm107a", 256, "e6e17188dbc679530a8bbfc0ae6e33f06a250e2e623776593a370a522f5703d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm107a", 256, "8aa3b00009ac20bd89a520afb7b10b5bec609df475c0bb3e29ee7e14a4270266", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm107a", 256, "42640cb5d7723b6d664e16692bdad65a633416a8390ff27ee1ced79aec126f84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_rgTma_clmp_sm107a", 256, "c602d2b471ee0b9f009dfff0f1d6edfb8909bda252f728aef38cbcc8c8c8c399", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "f7de57d6cc99791f8105d53f01ad7bb50c41f070c99367067d79b7de869d5476", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "734f6e33d41bff14014647b25687488ed0275a0195ba9b3263d4552dc23feb6e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_sm107a", 256, "21e26dc689d3025c247887261421f8c57f423a64d038277c01e42da3b812f848", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "039117a23719416c49fd4d4658242abea18bc9835f6a8ebb3d31757fa2437285", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163344, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm107a", 512, "e9be94b3818011ce78835df0ff6b8958fac17e39847a58532366e94fdc519e71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "9c83fa325778dc25bf0d8db88b3777729ce6a31c9a454120dbde90116512adbc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "58a349f952a6cc48320251311e0ac3c10b109406788f03a028c4a415c4754d7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162320, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm107a", 512, "471e853b4ef8fe3530c3c7b7d0605a2f67c5df86e053bcb3b24273bb4e4c5466", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "ed3468fc2d0549386f7b346d62a65c6ddec362202a4b85cb74ecd3333d2d2445", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "35b16bdf0d029d2a64a8619164a48df00500a39cf11d1d8bdf8234ffe432d618", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "d82403793dbcf58b8d38c0619e478d489997dec5de514fb2deb03c271d47cad3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "884850cec1d8385dd836ae6bf311bb1bb9dda80e78175a2ed9adcee6fc80e3da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "4c7b5ebb54026df070f6f4bf5c13a55e0615361bedb3577f7a4008d52620c889", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163344, "bmm_Fp16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512u2_s4_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_sm107a", 512, "b72b9f2e117584f49fff1241cfdb1ec583565d558994388c56ae74a9d5c8c24b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "dabaacaa14fa47f062576e6f2ad5b1c5c661969380e83e993eab09f10ca45f11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "d68d25f1259b5ce20c3df7f7827362137aac17f2fd7362fd2aef12c5210ea185", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "c2790f48f7cfc65f412474f893ac888bd28021eef37cc6b27de3fe285853668d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "6404618f2d61a3899a8592a4c4f16e0fa2aed0945cb9081f65ec770533bae5af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "803177114b823ddc91354c6d3c1c8c9f68d9dbe9054d82e27338e45efde6ad32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "0a169b66e024692136865cebf8a9e80f7ec94103cc9a2d7078eac70d2d66e5e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "a30bfd7288e257a4e7a810fc513583ccbe276b87e5868912c6eb42f7f8bde783", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "5ccb0b50fbd096a0b5dc101f81c2bac511968aeb35754a6dcc47c01d769e0906", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "93425551e5a3873ca22aec18a71e39f3521ec535cd9eb10d8499a61411515732", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "e352bb50e45e0e546d11f712c32c59a811097b0af84a769b97148df45e6ca2dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "0a773dbc4609cc4908668317497e4d160b0caca4c0e18137ab7173b8a5b51eaf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "c8ce6cefaf9e5fa07927aaa28c940763c1b51f1999749fac00d14b39e6c59836", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "b9f16d837308f7a7da67627a71cd85384c6d4dcb607931843eed8b5baebe93d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "8aa42f97fac7031404c6e43544a86242523a3b0a99fbfd7141214734ae30975d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "602b15db9d807021618d45aa7a157c18ab7e6359d6e1f722fbbe762bceaaa682", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "b845c3e003b8abf52a8c95bcaff1614f4d47da028bc55a6254b545535e466a6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "385e7277f79e190da7786c84add9487aa10ed80ac97b5aca666948169444806b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "9b6c47c36b76d4b542cdaa758c96115a51cb137e9f2cda475de202027dc054f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "a5b514631687edc771dfd36b654891cac8728cc5543a38951510586eee22ea2f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 77600, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_sm107a", 384, "b5f0e38e58021edc4897766d8d55d5076de6463e62f1168a9fb4f74cd47ab359", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "08884ec7bf7ad027b842443f329d6d79566cfb57da9d30149502acb4325d2725", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "f778539d2350a96c9e1e1016a9c3f2fe77cb79f43650d0a7e63e738c91d77d3b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81696, "bmm_Fp16_E4m3E4m3_Fp32_t128x8x128_s4_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_sm107a", 384, "4efc5ec29c47fb711d45d2beabb24c202c0504e410c34ea06d96cee40a7b42fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052679)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 0
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 1
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 2
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 0
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "10f566c9ae73063678f1b9a4856941f0bf2b4d454bce227a96d7a0eb305b143d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9599c3a75cde7673b95b7732e36d40ffce1c675bedbf300d0951ee6d685b11a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "685f01ee386640b0b503dc7219009777873622e13a32211fca6afb6d07bde069", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ca59ae0d18ebd579bc07a07578f3a095abddca5f010933de5bd786c2ed3b70de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6edea45865b3f2231dcdc1c64a7d9b2ca409f53b8916c99389dc3b886db16579", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "be88f1a7e53e096915a9b74a9f8086f9dea910a94dedab519660899c4e77c55a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "34c20342b99701c1bb87e92d01e3d4b267eea41861bb239cb15bf6d51062e7c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "11f53de6d0b9bd71b1d5478b19b83d08da2b62e433d1ed9443f65f68f456d31b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "937bcf988563d2321d4e13046a45f9bfc29b352e8c1958e4dfa54c1ffd9212c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "258347bcdf0d7cc319b5875d95583a9b83f961e823afabb986d161ffc78fbc12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "e7306dcebdd527940359744d0f6122eabaea6cef8f25707a73d7db2cb2568ea6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "db72a87c64112d78f7ae33d6585896eacebccebf447045d27c5a77529cf31bb8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "d3529bc0384ad8d22cc4eeb4f3225530d58e4876692b437b15e8afd71ef20be7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "891a1ea7db52f3b01bc177de9cef97a5817c308fa12c37498e9fcc25faf528b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "dba68b864e14f173d4933352021b87267e5bb005843798be4a8ceba84c5ea6c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f3a74e32f137ced970cbadd9b2ea84de7e15f25fb66254502f393493285297c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "c3c257329a9544a4117c5e0b93fbcc81ea5b27347cd42a5bdf037e982176ac68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "f97bb1de41e57bcd4555cd2babd96ed889a70906ac458b5836dec8ea20b146bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "7adb4b403e66d47085dbbbb0249763087b3b33fa6537ac95b140206be3a13b05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "528f7e32b452f210ec3986d00c72b5eb628c5042126525bdab26fcdf2f1cdf73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "4c6c484bd0fc00e2038d24e8416223f137b6d673acf5339c9a3c4761fab74f35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "104bfb7eab567f89097ecca8340caf65e32c955e95b1c390addd3d22486a7d74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "7a54076af4d78abc53823e724eb45ff0591d78ec15c22372ea74baba20c4b302", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "6097fce701c179d12dcf4aa5a3d92eaa89399af294863ce84c2b85f1ee32dc54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "9b136b387b19f18f2a2e32ce44c532bae28edcfe176c6169b12aaca92b98da08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "b2869a0cf85d0da948826bf7877368669b630b17850b11d1dfe4e3638f4477ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "b2d6e2899ddf9a8ad6582f5e6845319f96718ac79550dc2cab5ed3a8f44c32f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "fdc54d40e40189a3b626d8faa70d4e288c7898b55c06f4dae45b914f18329727", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "b7290629573495821ee8c9f905812b8b7d2fb49400ed644f7f7164178f310292", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d2e5b25c60d2047bacf2ed8a353693fcc6771c73a61c1a56b69345e30ff6070c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "22affd153404d4a06c676eeebc07dcb941de54dc85acfb8d72abdb9d49f1847e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "4521637c19d0b158cb06d155d0cdeb4e11c1f9f1ea61b3a3c95ebcd69ec21065", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "4c3d2a118a3d5f25dc65e9b5e3b3293de37ff97f5af437c7c651291c11a8ecae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "db2a2b587dfad7585557fcb42c8a4add47ac1bb79da8bdffdb829e8afb198d1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "c55275789421d193f713a2cb7d61370ae25c5f77020fe0b93ae3e84f368010ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "cd9597050296435116112ecd7a9c10364c35b5e3ac8b4c67522d7bf177f1aed4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "508b204d25cba90f2757f7353ab45ad9e233fc75c3e3942d502be52b308debb8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "a0cc3f7ace91367d0d44e5d041afecad4a02f9f4229973d4ff86ac47ff07fcfb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e7f0dc16969164a1e311ee6122c8c3a80f7a1d124d3a998e6c8cee4473e12d76", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "07d1c2a0c6d685c5fdf7db9b1d70e8eb123168f45aea8865130af257d25003e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e2c5c17d9c43c5b3694f028baa08d8d62b77dc0b92a0850f3ee56ba40d1016ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "560e716dfd5cee7ab6bbb4777dbe6b86af7eeee9603471829fe24b521932c207", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "3ef5692a4fb9a78b41b3cc8dfa0cca5af56069dcc409732470526b7fd86cd7af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c4b01515a20ecd5f9052b6e53fe1682903a751fd98134caf2bc6760e867892fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b5aac084d50989a918a7a31b5bf1f375df3c8d916508eac95da43afc57f40503", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190992, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "4322ab572944497474cecef16c018ec5a55ebea6708b72b6d2c4435bc761f657", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3f04e736ec5f7316918aba170c51b0d7ad880699ef155be990fcfc895515b951", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "bb3cca741c2abb005426b63a6594ef536be0f9affd3be26310fc6b505e0e4912", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "dd333d907359582beb85e34a9077f7334d2c0d23b339d7f896de5111cadf9111", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3719058dad1e6ea9df4767adc1389b89578fab41a70b31855e906a4cb2b66d76", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1170e0ea9987cea65b69408be5e7689ace65d472d4b4a2cc9e399afc6caffc9d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9cbefa4cce46119edbbaf693cef75a28c8631ba15c2ba5c789e556a056dee2c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d53da9fdcd032f45952bf3f8046f58b8a8851c811f15ddb7548b40b7e433ebb6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191120, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "fcb5c33e995ea849771f162a70c1467bd998a88549b341916485b5c641918796", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "877d5b2581ffeaea09c14067833d7c080bd2cbbfb0d97b87e318fe479be7be7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 198160, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "37d9fdd10ffb4a20fa12ea21862bc9636a6a446a9653f8cf0a1ece6ea0f72752", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 198160, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4faf0919d6da2b45422d1c8b2d02d532e87d1de42e026e1ea200cab45b694ffc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ad31dc2c056cf3d23240ff24fdbdc8e8272c5878a09a3f0551edc765e9b54e72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "90cb41d44c2a634d84a8f8ec0384124a20d3d8c5767b774e3daddf5a00d65151", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "bf42273d496b5b62fd08420e24e8dbaf2896f5d71c97c5039ee1f042c8ea00f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "252cf9d46ff62af26301bae7f5c96d3d2f6ae04423a77564da456ac2af86d3c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "64fc09a38290af011d90e7ef228530e0c160ccc3001b8d4b9594073158768d6d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "5107778361f717797604f18dc0cfc4ab67129d0560d0a583a0a810300f1bf047", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 198160, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "adad476c5a6d3235277305891dce6576a73808446ee4ac92ca450b45d279e165", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 198288, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f03b2e91d22fbb16a805a2bcc7bb384c1c6714a7d67b48da11982a802a964e76", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 198288, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "fca75d84e1f3cb95b616fdf210e7c3f47be9b1d8ebd3205a46a5db4f386807d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "6ce823a848b82cc81b3f332dfc647a2039e2835ea9dae4a16f490a3269b425b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "16471c3ceafbe16b8381ba2545a86f4d8416d2530d994148c127b08416aaef12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a881f7aec1b5e7a0b217c07e2b0349a5460dd53c40d710307aa538106bd151cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 198160, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "168e60a8b411036b844a30604c91c4840699461c61f869844f7f07050c426f23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "961541782a13468d73c014c2c62f82c0c750d4be2be336b7c07ee4a6cd65ec4f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 198288, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "9e37e33600a791aebe1d1209163c1fca55eafde4b3f9f6155f9daca9a5579e52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 198288, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "3dfa346939b53219e08db518668831483a9c7b4cdb139dbe76b91e5252cbf7d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "811aaca28b5c2487ff32b615d7fb77633fe4ea69c8961585415ff1c8c4d281a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "dc3376f7f11f7cd246f50e05ebdba9dc07e7e86660f062894490a5554846e9e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d135691dc5da5671abf70539b9c0d003b401229a816764090d1cef3c4abd3188", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "52e51ac0806d481797f660051f7b45a59d06cf251007f56d97c270713e38577e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192016, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ab25497da1defcef4a04c791459caf5ce196ff61917a4dfc931a8d0a0b678659", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4a3773befdcf30a8f1f54fd9fab8884b03e647c339cb88493e197f55fae295f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "db74a00e539d7e5905f5098743d85d1b077342f93c05d108ce100a731107bcb5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213904, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f8883d8ea4b77be559c868b31678a338fab2b3a6b3d1114e34e75e85b9e9ce5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "55faf6d99ca430fdb97425606dd08e2464e9aebba0f4672d0c74a56ac0784979", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c59562eeaf995ecfeb503f165e4ac91eeec3d4e5897271d5a4857aa5595d2223", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192016, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "33299382874473dd993472c77f25e55545ac8a922160eb04f0a934c7d34a04f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192144, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "53fb71e8363c7aa7b341e1c908b7808b77ac8246c0c2a66bfe3862478e0574b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192144, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d8c850838704403356d9e45c7df5de2357211de8909fb84e8fa8b9fbdf795761", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c48efee33a63a162cd072252573582a928efb04e3f5b34bc16ac717638b10a41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ac3ad1037b9004037748b8008eed53937d433e314f624d230aa36f607278ff2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "29b46ef423060ff665d93a05e10bc503289a1d68609f0ab57b11195c72981ad4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "da13f631635184ae69be2cc25eb0cdc56e53b5f54adca8595f345fc5fbcccc0a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "3ea86478c1ad7771e4c5d7bbe3b4eda84d0608ee98915a2e885ea6185ae24500", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "97cbaeeb3df5c9e840e27046e7d557c39323728b6c500c40b3a645b98278b57e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "eae4fe345471bf4fad7be4dab05cc4ce3cbb144d31dfa8f45e863a57d60cf9d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "ad080d820c898428ace5f9c6dd7033fc7c363d14945eb245c3cd8266d62ea8aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "2accb83f04a47b8f2d997f08611e80d535557c6c181356b982d48329c7c57f83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207376, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "939f186877eef0a70d139a238574e8d4d9248d0cf8255a1460c5244a5219b685", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e84651af3c2f5484696bbb658f7ca59bc6547b7091aba4ebb205fee623bd8998", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207376, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "07c7e35a5bf6af22452d001dc9b89f759b160a3cbd7af57013b0bdeec4386455", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "c7191147dca46f2317b32a85333c9e6fa136f58d5afdf2caaa760781d7f0fe30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "9deb8b2640784dab8402be110aefdf71f6c2fd4319ff0f23c673b27f7b8dab6f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "22ec5ca045b2b2266732f4c1c085341cad53376958dd73bdaff6e42075ec0ddb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207504, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9ac31f8d49ac6663817c54c3095b9ab9a869cbf31684a99139fbaa43d65b5a7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207504, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "45eeb641bc1ba7190864ab5dfd7969b0b0e8edf5a90c98bf44c355c909505e69", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207376, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "766bf5f24dad465b190f9ccf3bda9ee7b96c9dd23351ec35c45e9f3bec542310", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207504, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "940c64a708f0aaf55b871bd2d22a2db679233b9bb282630a425cc7ee657630de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "31d273d6346298254ce2e83675d9fb8db361faa78c13443e4cbae01a0f8042ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207376, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "46d6333191a12182f96ca45b86beb45e78f68ccca988dc500eee9c2275630af8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207376, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "25f2ef4c3d41124713846a02313d8c71782ac4904d16c1b331c5c3dd941ed24f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3c8c79c9c4c2b0ddca050d0f1a682799dc916dc361d67446cc264cc28040fd81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224784, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "626e65f46f121fdff8222bfc8d441ea5c9c7dad990c412031ad7a5b0ac53bffb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207376, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "120d6867248b27c2cf005515f5f5f389f9c0b79881fe7d7ac200162987489dc4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "209f185394a924aeedd36772275e69f114def7e98bea28d8090923b3e017339b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207504, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "222e9a358fbc591e3431e8a6b67478a19ca28ff247f73ad2bdd4d4663ed9fafd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224784, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "41dc62e111d23da85314ba30aa68658be0cb9abb2c740f7f1cbbf4d863ff7632", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225168, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bf7634527afc399f1d8b4101ef34ec163eccbf5f19380a3218f0febe9f925802", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7a86d197a82e8c191ad522c91210ca3e2ed3536f904189adda54a4f869570920", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bb48fb037b387c3516b7b9e9e3b20d102d52b1b8485cc1821baed8748accea9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "43a639f4e655ffdd602992e55c18b28d1fa633d9cb4f6676a91ecfbc09851855", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "56a2dfbdf3186df6194eb248ce99b7c36328bc96a7cdecdb0443168255633eaa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "322121de8da2340e2223e3d83c0e97f580b385eee6fcce5da366df67338a62d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225168, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "02df74c33a1abaec584b5b09b588d847e6e3f7696116cb3fd0ee7f053baa3b8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "245d6f55429f70979135073be8334505b6960bb1547a7039f8c4567c78b25e50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "26a3ccbecf178f9e5a0485d8eaa0af4e5ebe09b5acedcdb2b72a6fabc4f8da55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213520, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "5147827e4da4406452df3826c7911158bd51f28ec107ef79a209302aa3594fc6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213648, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "16c354e3834e3964dfac3a83ff1e5f056b0afe3551d7c62c9aa16734819067ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "813de39a6280c30258dab41782b69b04bf63241c1049c375f1c6a57ab35491cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0bc1d7870270c25ad8174b543529d0db4e8febdddee51ff9a43ad116352f2bad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "5c32c57d8567bafbe840a8009aec245883ccd2189907f7284e514c7fa2e6f3e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213648, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "622a270fb1b97ce872c4a713ba86d94610d42e416fba888302b0928c02acb846", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "6ccf45f46dad173cf524d3614d088cbb736ac9489a8605e1e118e5aa4f4a7b8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "35616cf30b39edf3b62cf9c93f8365869a06372ce12aa8f502ec6ba81ec58c0e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "03a64397c92efa71d63a1a3896f49795c0301a2d75190f0e70f4c9cd7637d92f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "2c8adaa058e703f3f3fd743a15958505e4638e593770a238fec5cec79893a0f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "55685f5fe6dd3c62e87f0a639e84745b956afd8da6b32f3fce3e891012abd4e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "844359d6124f4e9ce5c169f8f174c0d6dbfc00950690381f0c82f68b31e94b0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "d7eb57d3f3e52082f1951c2bdf69f64173190da541b140bf8eb35747435f2512", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "64ace7833f33ae0f3d1621101da1ee833474c950fd0027dfa0c8bfa5202a8c8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "3aafa99480a5a742c047e315c46c2c1a235ca2f20ac32c242654ad25294d8547", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "1ac082dc156ba78695478c23a1a3fd63c33dc775dd7de7643878dac0127e5f72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "22308c308385ac7461a861b54ac3d56d0e0d8579bc244d9c74df6ce1abb2b810", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "372db497ba0c0d1ec06e003ff61d413d3a136d0bd3ce7238f1322e90e5e17674", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "19720e2c838b9ca9076334095054da2e4c31844b6d8845a67fd297af7bceaeb4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "3ac88e1d4322602632a63e80c6b503c5010b2480e634a1de88d3394b34877879", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "a58d84e19402635e0f3be394a5f819e49c55d60d33667f856847a81ff0a1efe5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "4e83a1b1e4d3d654457143ae163e39484172fd8bc42e4d937089ca5a2badecf9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "0c036a5458d8e6527360da9e3a707aeb3a15943ec2e6bfa3d99aa2b9735dac9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "fdac49a6759b7ef4b69c8d15cecdb79ce428e18714569c8a1a42eb749e6782bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "4268ed9e5c9d3d1e31b9544bdde02db0651958c227bd6f2ea1a94c5e12ab3e38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "93df9220ed546cbd81634908d635b7420bc136fc86bf7e3946d25b906f1cd51d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "3f931af4890d3fd378aa9d2a30178c70f38af4f34d0fc3ab0ea67c0dea6a7ae0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "d27110e4fb47db3c03d67c9c1dd84cad8c94de9736e4e1e10546806d9a4b92fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "72f7411d18d7533df27bfa029f4fffe553b568ff8c8cf3b7d47dd4852b0e79e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "cc02e10b6b8858eaef4c628c02d7c097594e0de27bf7cdc6745b54ae72412b2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "5f4f6968a8f9ece32fbb12163bf3841823ab28cd5a4f53cf41897baf2c722951", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 768, "523bea5e08a6c4c19c0715f1267751c1f2bfdb5d76ae37a9bbd94fc8af6bc221", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "d954ec8fda68ded4a6f7734d13b7cb1b654c2d63e12e96e19b606c45e843447e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "4d77944d4654ff1b00d025d8242b8ff1740a583c9abec86eb16e371322fce3cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "a53397408c071ff7e791c95ac5bbf58dfa721943662f228584fed42a12256b3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "d0c90dc7afe80db5c071f45a2ec6b58c25fcd942e6dd67b6b7bd5fbf18d25da4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "df96ab18de309b6234e7b955ac402a6e52f118d82e366e2e85f25397c12e663f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "4d269dea5a8ffac6cf431424d73b7ec37814d768524ebfb019f83c24732d263c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "50a5191bd2b865b6571059cee944f0e51890d10759c7746a9713f3f39fac4223", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "8b61042631db98574dbd24e882d7de0bb1d543cb8b3b849e1093602f8c28543b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 768, "d3d2b50405e4ee49a46a6c3d66fe09f32941848e2e23c0b2598ac62a89adec26", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "9bd337605d877a197081a1489e1152fd19f4eb7f9d56d6323b9c58203315c530", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "295ec41194f556b062ad022241e9c8f41d4163520b04fc7d14e8370138f3d74f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "a4e5350a01d03b79bb71587edfdf0ee5e7d03ec5b91a786619a0eefeb2a43dc8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "01b0fa73a20b131c6f8ff583d1ad398201e07d853de35a4ea898323784b6470f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "05157c78336356ed4cfbd422ffa5634255709783fbffe55869ffe2fca53fbf26", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "bb3f2feada6b4f6adeb37650f85f815a597df5a3372eab37a8f72628c16705ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "f4c09709434444d8e29aa4aa6bdc6166796c599bab8efdf31311c9bbf0604153", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "18cee6a6e506ef0e625b2805b49580b4109e8b36bb3738a957803c25ca84f3b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "2ef6a104498de24f2c317276ca16ba65a91bdefe47c1c77ff1218b4be7a73a4a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "e291b7f0340afe76551b37ed49fd8f39bef97d969ce2b15ea1af1f8872076bbf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "cb1e036eecfa5f82b7fea7bb4c46c2a28531977bced0b6b8498c24374e09a812", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "6520acf7fde9f62fd0bb39d23dd9fed145df199a87d876e2005c3d142b349207", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "25e2325efb39e612ff1fcd800ab65065678fefc61966183b85b3ff9cfe6dcfd7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "99a8626dd264cc5cd9f96031b2f068b8bd3a48b3ed8a82f502355f1dae5e1cf7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 768, "55d2d9a51013ef3a91a35d00908b8787c79629d3e95429b50e233c1396ea5db6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "6b697c323822994c702230b0b200726b9f5b73c2c1f579d7f0ae81b02364ca58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "d3ee5416ff31dda58b865e601e169fb9e2c5a3b925c417e273c04ec6ad775586", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "b5378c27dbc676689308d5b77ea3211408e7a3b8c24345b24e7232564716b7bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "03c04b5ab8f5de1eec32b040f440a547ce131e81d4fb908cf5e13985dc0b08a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "1f85efdcd70bc624023304d4ba197790433db0e67aaba46aefebf5b187f0db92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 896, "b2729fd08609134e70574ae12b373768866f35b2bb1d66754be5e9d1b8068131", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "d456e2c370320cc9ef92967e68fef35cdb428b8d9a9f1af35904a088e826e7fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "aa1471358e4b60dc5e26ef2fd25bfec9eea8b00bf9e807a88f27806ae369b6e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 512, "180f8bf13057061fa6839e1e1a1f36deb419f22b0863342d0f3cba9e60381ef4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "9ffd077eb9cb4a420e72b305e62869d25a3ed6de27b9998a8a256f53760c251b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "7a821333c4b2cde76d12aff421a1992271b7f086afdd23c8fa90a6b9032e2ce5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 768, "4b485f57c969c9f46c369e944e50458ae3e7a93d47252bce3e2ed489bf257cb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "8e917e873627c1642aa6e703f360917d594a7740e5f70048eb8088afd04fda6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "f8d5ddfddb608d1d5b56654ce9e27a7a96697ce5cdd5bb2faaf267882526a720", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "1e7cc811c0daa11299519e5c5a2d006c57a116c8cdb08ca8557f42eaf202a714", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "8037c7fc1981ac72d963dc3ad9462d44b92ca5ea30e72393dd02e6b503d29254", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "173ab448783753642ad1b4cfee1db191c414b029033b6c48644dcf890638645a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d9bd2a19855dd21f84362f552aeedb1a15b557546827c2711bd712d4a99b0f22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 896, "775e734023dfe5b438dd6f4473495925dc072f0520a65113cb36f73bf7b496ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "32c264273181e4b4f475e049a5a46a0094a2b669dc34ee5f6fc70a6fd3431fc4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "de196e815161b5ea8f2fdbb2358a1f4068444c563338be51315a30b5e192e4f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 768, "aadb7c8c2ffb2e3e6c0ca1f3536b02d46af94dcf1d042bd582e272c24595f773", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "3540f10d18834319f37e2a664204d1acde545e7ae2754754c7d615a6f7126fb3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "9b5841246bff3442c255067f897ec759e54f1f94243c1476987b02407d42caf0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 512, "c28bc5f3a3d6bd267c1179ef05a646dd28f583b334d0bbb54c978fe844838ad8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "543950c51415c950f653f4a4bdc5da4623d8106366727e25414bfae6c0a84737", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "87b3783764489d2f899e1f305647d839a4c9b60e7c15baf102247c70d93fef7b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "835d6a8f00746e35ad1c0bba42c96fd5471c0e9c08f516628b6478bd7152e7ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "4e06a3822fd57061b54128845e18e4bd80e4e548324e17f62980163dcdba589f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "333e97c4441562c55fce0dc3ddb8043f4f9dfbf41bae6c93fd98af16fc6feb33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "e5f097c911c25eaa587b2a1bf4f76b1b61105a1e6024df1635dc8597ef3ffed2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "cf3286cde67508f955afcea54b1634c90a01f027be41309bf849dddd3c61acd9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "3b6dd75649113425db053f7ab62d874933db12c97e17555bdf8461996936282e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "b64db36ece62d00a279a38b8573b6a1449680bcdd40da008702c89cb1ea1aca1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 768, "348c27f39b26db7385bcd4d3cf264e0e5097e4d674a810a223dc2bbad4000e78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "aef70c06291e8582525ee8942fd61639b39c0c7f7afc9f0f3154a2ae48b3e588", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "cf2c2223f96e99361d2e00acc6a276928c41fd4e25cb221c7ef51b9f1b303c49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7b659eebe286cb861c797647c67358fde2f0b1e1678c30478e7b2dbbc3ff1515", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "222469e875a9e86c4e9835ec29361b6dd8d50a8c9bb8e9bbd91ba8cabc6cfc41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215184, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "06151868f4c8287b19a19652838a0db039aa8b3d2e692ed5d382731f339fb85e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "e5425deadab817d736f27645bd5288474987837e3413434c392dfe79b8df1037", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8c154a439d5e2186cd0634481763714c49a92d4e55851f377695666c3f4b826b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "ab3d20ee8e645a497953bc3ef30a051726e4c9b9b0c9eb04613f97144dcf0527", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d0e2174370cac34387df9868548a3632382af455f5acb53da7d02075b5f3be79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "4d7a12507ca590e1d2fc0d5dc84f2a9b10b354ee3a36ca8dd40c39a221eec2ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "2b84c5d8d71c34bc7869d4da4c91bde58561c8af9755f1428974abd8cd233f41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d9435bc43ec952a5512fbebba1491aee12ab2a1b1fcc012a6ac6b80a209a53e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223768, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "4d8b18a3686925c8b42be330d39573c0c01b58202496e7cf9c9ddac24cc9a1eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0bde2875e4265c112186fb24f73033e66abe8ce657e616234b026a2ea243f435", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0268435d2c252e63fcad1c7e97792470a844bd8c2856c203f69885e0c24fbc30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222352, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "96d16c73eece1d64c02b3c98e68a140460844afd567e9865ccee6cf9bc776616", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "129d24f47d020f71f45e838571c65e14e6308efdd64a1973c7245bab30274292", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "ceb05f64d21bb51a3bd8682183ff36836f2998109881058d37e5d480fa58cda1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "269b371192a6f1ec132c7360755678314885cf5fb28a4c2b5864c602c40b39d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "59f35ab950dbff9956debfe2207873a1c4554c78b3397104a6c7db4b4f680c92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "fc1ff7ad95d6b25a95fbe4fba104fa9c7dba728e7a021f92ac61db50defe4d9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "75f9d87364dd0861511b53a7595559a88a2f509efd9b418c686689de0e03abd5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "32a0ba239efa2932da21348aae8375038defc3f2060f4bd607acb1b27578f2d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223768, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "34a4b89e0a4769436974a2de3ff94e72ba68fa8c26497737864fc387a1979e94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "9116f17988451db56be09ba6d51f1ef9d06562ba1db7de91c0a9e9b63c12d0b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "5fefb37f320809c8b972703a76a1475f22759c01da461d808c4ac908506d2879", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222352, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b822760729d07a831eeaac44186684344dcfa1b27331c3f01830955665a82521", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b20491ba424b5be155d3a355144f6dedccbb865630ebee0c261a07d10cd8f4d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "843875987ba87eec9968e250940df9dbc817b4ac5d9f0d8677976cd2b6b3b086", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0b4fe23567586b1121b8f273050fed179f8a0f4833fd9564f21a35ccbdfff5c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7b8e69ba69f90165e805ab5b850d50b9c4e9aa8d92888e696112e4488e3c3415", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "35011b8186503c720422835ff7e1691e23feed94fbe6a4cc3bf9053c5c66d7d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202256, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "373e59c44f491a3e787b60d6a11ee628991055b12cc496a28c7ad112126620fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183312, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8bcf52a73a79a8978ce43f646fab5847299960b87c4f8647a7ea7e38167d56c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "e5320d749ca4f90de036f6a57f359a8ed07d68e1e20d846d587c9c7ebe3d4450", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216080, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "35ca4e2251cfa3224e2c6e1723696c714db419b0a8f1b95d47198c2057578805", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "1186faf1d131d4c88061a37a01129df8c95f4dcc3dbeacdabb3f34476e6041d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193552, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "08955a3f148ccecf5f56d2e65d8f8d93ff1c4a254dbb8841e9662a20f03fc2e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217624, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "a4f8309122bf7244f4d9e7e88c4499a7fe326ade25bf84a84e460d023bc2048e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203288, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "1223ada735cdc8a3a7a518a1b8ae0eaa2a73394c3b4b0d9ec7ae20391b6dbf72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202640, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "d5d938deb9cfbbaadf310ff60081d4ee89940a03621ec20bbb203b1a2068f03c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183440, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c54fc2eb2195f14f663e7878c02af16a075d974fa9f56e7bed52f2a2912524aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193680, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "d751f22fb03c0f1998c67adb7032bcb6f7aa0dc60905a69ff5b6168fe88199a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216208, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "41ee6295e24e4743005f606f6c97b8400e13a77054d367a34623844fb2862c74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "0bebb860ac006113290eedebb10fd853780428ca623b403660fa9a7e1650404c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "3715777af7bfe449c7c0bb539efe1f7c315261e139af93f87450998401ffcd5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216080, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "50c8677c32ddc5186252dfd9601e64def608659a79be260e182fe09f3b3c0009", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217624, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "5531860b8545104a818a847da6c3f9172942529ea3f7235c20dac9cb33156a7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a65487b00730be82916715c99b19971f7102bf719b188893900898e586ae6941", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "3dce4c3c71901e758e28281ca962804560d3f59ffeb6f749e7b1e6babca74261", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216208, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "4868d88a813726c97f59fb17649df14b74638d86482838e513f53ed5c5dcde4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "5534d015005f8e98c74a9bb8aa52b9d2f44528a3bd288c914a2ac59d98f8953c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187408, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "32177b7c9356821613ab7f40b86daf0928bf318143a55d1aca44d06265e49221", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201744, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "f44a86246d8a3d5000db11613a0da41fdfdc988cb99e6d026b2670a05f6544f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187536, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "844f01e7f3f3027979fa1267374aec7c2d234b7d9365b7ae13c2550cf29405c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211480, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "ac9a069652a0f6c0b2bdea17ccf71ca6693617f0ef3b652ac3be4eb56702de71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bb9fa4411fb2035fa090d319a157475cfd2a14c6e571ef8a45f67c5d2098336b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211864, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "4ad1b9e814b3af0767f2fc5e1be98c08643caabccd1ce0d44a7d4079a0085770", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187408, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1c2d6bd5d36f0584e41ab3b9974b576d2f8e9bb228692f46d37603a961871b38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201872, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "9b79318d6aceb713d62f976bc47d110c67e4faaaf7c73616b8ba5801ef15c1de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191000, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "3f0de3d1d937f9d2e2126276b4957e8f362215d6e2451ccfb38b47d0d6208b84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "405e1dc034f676336f105b0f5a0176774d0f3dfe6c1ba7d8a222b4efca81179b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7c1d82b2dfba0471e546181200ab6af38e9ebd5631130df9fdfd5a80aeb38076", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187536, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c9826c82ab9aae232d3793579653f6f87500337e1e4880615c26da106be621c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211480, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "9a9ecdd73604da1e00f7b49174dd628d33cfe9ad978d40d1f658a1ebbe0c7610", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201744, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b0e953a96a9c37897b4e90b0a807cc8cc04dd4b7d7e6afd904ec9301616c4967", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187408, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "2171bf74fc45e0600a06455496483c075add60cdbf0ad515e21307e262463583", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "df7a7b6aa957be90c80a447992d9866f7b7d5d33700acceb82acc17bdedf4364", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 208016, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "ebc19bf1c21c447d6bd5f0afcb8a2ebd6dcc358e85947534661155dd50fa023a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201872, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "e5632eac5ab4fb7ac3ae09cdc37b8f57273c052248acd9e21774f2223126c13b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "f57d5cc7db1ee4e64b02907a5aeef0244aee63fa285082187c54f764284b5f77", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191000, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "ad86551644d775ca8c32a40c796a2ed1f6a8f89f2c015c8b39a90a821ed26610", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211864, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b893a8ecc62ce3eabc4763fbab4e434f4e33ba4d51415779be2b84363d631991", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "271cb3fcb585dd3d3c5c1c75f880099a8d338c6dabc3adfe56ba642dfe121d49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "aa4e6de220ca03f996482dddf22172224c6064f5ef86d23389b7205c2b4feebe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201744, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "db1a1a3ac4218e4fb5bdebcec25b5c666650db52a327b18fe9964e5609d47e2e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187536, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "92bde4ce1383b545ef92626fe2210a125123e1dd9a3c551f9f7d6cec15382aab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211480, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "d50861cc9e8a84049d85b23f4a8cbe44dd5ba28a22b2b4dea185f43a1dcaf403", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 208016, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "c632685083ca680a67a8307b2cc15dcbce14942dc600d813c38ca23323938091", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "9aec9f51e229f035cad3798f9e00e56b06e947d906712a07825eeacce2dfb56a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211864, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "e1e8cda87269df7b58b5207f3590e4976b3ea402de788c9289259b7ac27d4e02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201872, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "5dc6de0d466a0a7f11aa91a066025884c5999bde4d046aadd013dbd5ab4afbdd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191000, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 768, "3c489a0d7b93517ac3a46bfeb632b2195f6decbfc078429699dc0b9edfe3180c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "4a5b2b416b244e1aff815986e7ecaadc5e0ea9fea4a97b75c1d0cf419423e85a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "e7a7c824fb981c220c05a8b1c31d6bfc868819f6c879b994c07d0e9c6a18ce1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206352, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "3bde7b93f429830370c8a6b2b4a09d687efd58dff9a64726b15e83d1400504af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187408, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "9a7c2a4a738066e9b88454c40deba86e364522f35001e55f4709deb413498b4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206736, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "ed10606c99e45223b3dcd8e79dcf02cca3b62c7cee72d6223dfe01d6f4e978ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187536, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "a9d1af2da815a63e626486a4e135b55675376c8051530b0dccc91629c454459e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201744, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "ea7ea08b1d470a07fa0aea3bd0341d9397dae4ed78a8f95bec990a5e00c059e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211480, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "9928200a9f6f5790ca994d404b482dcab2bcc2a26335cb4d09d0b8f446f88857", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 208016, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "eecda9c0bb4f998a12935e5d4b72281704afde0edc5255ea1fe70a9b63b37d9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201872, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "c2401cb93184b29143390f0ade3bd4d315763fd7ff5f5bd2a4433460db777f1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207376, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b6e9114165bfbe2c91df07152bba61792c038c835fd315f727dca1cae4c4dcb2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211864, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "7932a1327adde2d44ac63e156f3d50096a92c2c7e2ca685dff5775f033e54789", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191000, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 768, "55b381d8958aeca8e7aa9b625014eb62eeaafb2c477beb6052cfda1dee576421", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188432, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f0da7ea95c89d533b37a00cff7cab3ae90537d7fe125a4f313f4d786bccb6b59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "a5c5a3ebbac6718f4bf7477c010436890ff6d81335740f58e3ebe8fe64215d72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "28a4487cd15fb22b2bee37507ab483741a77d87ebce926bfbace787bed353d5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "98d158c01d30be378ea3dd86ad67b7e5b119164e27c7c0229ddf73fa93bb50bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188560, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "91cff5ce58c173b16593336b490c1add5b0ea4330ee74c192a071a0c3ed9977c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "c334a48e27ce0be3492cd69f1a180f316a3db8d48f76e3ed48529e2ccccd0b0b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214552, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "a12c3c4cdc8a4917e36ade02a5ad6f9b8372e03418b28095d6306f4ee49944ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 208016, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "adfd4bad8bb171eb200799564ba458ecba41dfd084d5ee95f61da1f3e056b3f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207376, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9b32239b8b39d4becd20f910a62b102f8b2eb2d2e505599a318d302b7ee41d21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204944, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "5883c051aa5af0c3fdf821ef5d8a842af24a01f83ecef4962388a716a0352fbb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214936, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "2684514be22077c4fb90c723100dd6f67be46956be41bfbbeb65b5ab65a4d24f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188432, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "047c9da4834d6b934303c50b1a9b976762629dea870fab2f196122611d9ebab0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0d8307f2368ae4110d6c944a6ec04cd4849d87511e3d1aa211e950567d48730b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197144, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "23221357f093118c614aa25bd8a87f621288a6e0e64a9f15f03fbbf668b399fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "eedb6aa3fd9c1b708f7cb485305681eab3be2146bb33de3bfb7e0beda4244040", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188560, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c45895015b3f028c2bb5825f01afe99bb440aeb380b966e8f9f95cd750980c8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214552, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "d22ba5e0195b5448577190a94af39c49ebeff84b7cf1d8774dcee1dd5192ddfd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214160, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "8fddf58cabd2430913ab3c9b3dc47fbfe76d597fb41273f98c5f00c9e375e4c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197272, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "39b3bea9908b1a5a17350630beec40824c741359a471f38661032cc911c103d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214936, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "e79199c97edca8f7d8f3fc6d33d08927ab71052e7bf0f9da760285d25f9b7628", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "817f13b3dbd432fce7fac9be14dd7ecdd31b661bc3b2d4dceca6a027279a2a07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204944, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "52488aa0d8895ce428bcff1d62f78a61eb6a0dd3cdaf875b3c508b5821da486e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197144, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "3c4e28c00b2a19f8f2b8221cda4459d91040aea7eaa0e6c6cd5be513d65c531b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "38cd99f31d91b56bba9c8127c8b694b4c67c038f0795270ed29b8888f7e97a02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "10f7da36e75c4a2441ae79dc3a9c6a2469cd25b368be660b414ffaaeb541dd77", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197272, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "8910517b3ddeeff17b314c94c1aae0fc5a0f89deb3ef793d09fe72cce0afb5a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "e725d7b2449658928b2238139506845058cf5cccfba8d6861cf73213fc5e47e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214160, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s8_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "2b784ba692022af295441cbcbaf48f1be8cf9933da73c5ff7cb70b499a88f983", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "1f7d27858fbc6a56c33218796ebf3502269d24e98000c6b5fb0f6e161d71bab7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "491b5a98c38b4742a7d229865f342ebfa5b6826fb6a2fafcca256d753b5f4bf4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "fa83ae33b205c998a978ee17d8db64667deda2dca4f077489b16e6983acecaea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "19a9010fe49a42f51bbc0cd39fcfbed83e410094693cdc3e1436e991a875ee61", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "afd2e0e39e95acf92e10057d4d0752409464b60ec022f76e0821ece10a6cae74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "12eeccf3133d82f8b8cc14c4da63369c10531372705ce2e1632270e1336dfc13", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 175768, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "34e0dec7f61c392ccdda72680a0da2e522f9871077a075c54e420ab45afc5aea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "b1fa187bb31fc53ba434e921c7d1439470021cd149e566711aad9ebb3c65c310", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 768, "0e062333a415423c36aa3626f766f84472670dabf09cc979f6e03741f4a0e643", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 768, "e2aa2dc6d239168edf93ab1fed852149715549b904bf50184bab6402cd12590c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "338438064fd6c01ffb0f473f3a85c45d8fe6e890eed3f3ee1e448f78a90508ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "2403a6088da59d75b2bbb50c4d7c794b5ebae72b629e57306e4d878d16a0e577", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 175768, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "98ae817aef4f216cc632a9b2fd4d691a32dace2aef042031118ed0815014ac33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "b8c7cc583fea3e8baf15b36384ab43be7ed639c4a006467809ded50d8847af54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "847e15e3cf704d973343ecc4652a682f67037a232e345b3131cce7b65d8ec882", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "7465d793f57f6cdf5b526c7c5292846c45807eadd3067875bb11f45048368e9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "b984912b228cbd9e2cc182e77e2053506589317867d08a2ce6c973d5e6d93d01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 175768, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "d1ef7b0142a0302e97ccfb6da0d78ec6900628a196a83f1d1885e3c7f331ff96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "182c6aa6673e9dbffab4c004faf76c0694ec8113d9058cf7668a8debca454e8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 768, "ddb8d2bfebd2455542e823c0fb91117a3575a905c211651d231478053f214f85", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 768, "c923f213932b61a347bf21a840c7144100bb813400f5375923c3ca3941a7fb31", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "67a94174c95e7cf03d8871743189ce496c2ef12ff2cbbfeff824e0e57833a53d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 175768, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 896, "f5b8a0ce250750407c21ab89c7045ee1066d45588e1c5264f0909fb85627310f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "96421c75f448b8ee28cbfa2bf90910f57fd57a10e9b172e44958317928a2b34c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "eafbaffab6241f51c53fd7a801a9daa45bf4282b26171bf353e2b2cbb656c393", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "4da2b08b2dcea49273edf4d10440900bfcf8735dda978d2dafc08c0d60c94cc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "0d1e3b0c41fa69f46a5f34308a1f131cb9194617c46b0c4b55b598d19c617486", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "2f252f89f179414cf13fe3c9ec87c8af7f11bc744fa0ac189b6617b4003d0a3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "984274b883e3db73563fb447a3908fec3ec6cbe6ca7646373757ee547376e20f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 175768, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "05a49407d83b9bf3665926b7ea78fc986f6f3cfab6424ac28d6b56e26ea3bf74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "87d2f102d1c4cdd03b7a898c592fdbc3e2643f5542973cff435092bb7efd8c80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 768, "90a7855874cee4665281dd2d78967f07f1ea54e7653d60963ff8ae1c7a084dec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 768, "a218a1562904eebad667810ead7201a1d3cefb9dc5df27481ce10a6b40c9eff6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "942c10565d05bc96f000b6134e8aa12a2a98db3876d82c1f0db2c23b9e3074b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "69ac5870687b6e6c5861ef047b991df60b35ffb969c796d56f85fc605c901c53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a7af04df2f050b5b07228d4022e13be699d6bbb3ae4d1a7f2113f4176ebd51cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "905a817f82c880fadc2c6272f91059c087f7c54d695fe27841e5d40c78a42245", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "edc8dc221cff6f752cb5a00824460f49e7d7fcbe9af300972b76d71b4259f28f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 175768, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "1961e55ea8a9a49857c543427c6a1195da5add21245246d8060bc93aefb6689e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "e5bf0fa36a8f9753a9650f7b135ad9fdb61d18bff3fae216a7d25471accf80e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "a7da68c6092ea40662e2d914d6b3c90f56755d7ab69d9c6254d94ab399fe2f3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "a83b5a02a50270531ebf48dd290100412b24f7daea680c2ce92f8420a01cfe1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "649d851ccc81046887d3830e00b88553749d8f499ae705c334e74b0b5970cab0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176792, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a6f12d73194ba0312ee83de6050b4b81d58a4b539f2998862eb86967a014491b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "409021b056e9a9c3857668631b53df717d9eff22f2ef510b48391411b7f5416c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "bd6861187eabfa15b684ffd92a51a994bf80c9dedbfd2e1ae817e8aaa74477e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "0e2135bb36e96fe6d2a972d36ff169b73240bd92acb1925ccfb387e9475b7d0f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "db6ce092f3a5772568aefa8dd421672b68c5f13ba4b6e3249025ecb0cc0e3691", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176792, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "e83d31cd92f58843770ec6f4694163cbec2b453da2b93f72fa5297616ea62050", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "314bc0e3c9f5e104fb72d86190a3dae4a1091a7c77a1c3287d44f4b7840d6023", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c42379d10468c9c818b9c4ffbd6cf09fcaed6776cf7d7c6674bad8e2ebd90d5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "52425cc64bd543177cf57daafe305e3805194e39899f02a90e22d657913a0d7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "893fdf582c7fe3a369643bb8b8b5299e7524cf5c9ad29e1902ead2d1a9947865", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 640, "fb710aeb4da2f3a8c36fbf678f5f2adc939124e78cd3112606a5a6b717e99a64", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176792, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e776367327d950affc6a7e5adc3b2c3d36f475907f97584a178a49c0bbf711e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 768, "3062658b19d7da2f9069d6c42e125477a9142f5cc0045a729a27fd2d4ec1544c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 768, "505c68314c040721ed150f853f8bb6c160cfc9efc869c23d5aaa22106506fc47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176792, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "332fdd633b18352ce8deaad600d124c109e412d169f7c4cfc6f38bf60d88a6b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c8b2dc967fcbc2a2f87457f3803b4e9401656f3b7518a38db1fcaaad43ccebab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "64121b02aa4fa7e67be98be826e48a07d75e142f9441db3f0e1518bec884ba10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1079d03caa5df0d76c5b5b6f1025e1b515cf7cfba39d3cc9e1237046ae0974f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0f47f10e5d8c41e2cce8e665a108130bcd7f0fa08bab197e131668b9db236b4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "40f9a921548592af9d71e571219bb794ec9111718a60112090690a2ed1840b91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "6cceb374246a04ea16ecae80d101f94eee8303276f737c855511e6a99d1aa485", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176792, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e386b8923dcaebe2fb4ad0dd7dcee5814b127af3720e2e2d26dc42493915ce21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "951765885ac0fd9889b0c5bdbc5d6afc638de5d29d1736bb36bfa36231013bfb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "9f555e12dba7a1d4e687de35765262cb862a4437e2a2de43b0766314f578564a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "1882b36d556cbda0ec1ecc158801eed6eea2e90627693da2e1f61746dcc26ff9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "b5e4985d0b3f9cb319f88c9267c407f4ec11688651cd07d81cd554369b794b7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "95c363ed3fd22dfe6de345f786b4cb369afe41f52912be50df6b42c27e78b778", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176792, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "3d75660bdb5df315015957bd2b11f5521a44a4e29726623929ec75be9b1107a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222104, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "741470a60262b6ed79d1c3a47d7aea89cb2b0c18854a04f5f6c141d14dacfc7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "b63b31f3145dfa0b5e51403bacd4ad00a2ed5004dd2d755de3e2a8ddc965a74a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "24513b66183206f7cfec7942d50fe30e8a24495ff834744174264f7d76a392f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "974f462102f786a3b0430bb0ae71a01088d9c11ba596e40476cb0a4c5ef12e55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "8e41d7f101ad07a4813ee0b1a46dacb18a7b2231218cc90fd0180eeee1986f5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222104, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "6c608ff90d762902060696dad16defe71ae59b30b0617f795c0897768ab5dab2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "304a617618ae383921d9f41d7ae461b3d5a3d7594f9c5d6600979351787f155a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a36f25b3d0a296d24ad36583ebb5a885d84a5e760121cd0d632c8a79e2e78618", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6bb5c757b519e8bab217916b9b2db11fa341bc190162a755d57a5238155f834a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 768, "d5face8a1aae5fd80de8da0f966ed1bb5d88faf65fbf0a68c0caa8be57111769", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 768, "55fff281807fc8ca6a1814b58e6955d8dcace681ee6984cb6e076da616a4401d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222104, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "92cf975293e1ca6aee6efda800ed59671746723119d8f8ec3eccb1e1c2392b46", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "c5ce239f75afedc3bf921afdfa62ec0618d5df6856aca52444bf3d9503226cda", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "02718c7e58f75d2700ce97c418031e589a9211f6816a17417288f965b48d96d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "68dc1d4f3c7f363f87d97445c4f199c6e8d9635f002b6dbabb5491bc72f22a1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "80eadd8a1e10b507f4aa90b68833a0429c0ab495c01d91f21c0329edb9e9d806", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f71a134f833ca827dae56f35fc88f3fe775da583c023b2d5585a08b65fb51943", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "107c437ba522b0c00e9d3dbdbec978e248f2f0cb300a38309a2cba194cd6afe4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "18d460413068d22a14fb0f367dc914a9e3cc59c7c3a35d029e0d79838107c0fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222104, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7fd69df01e23ee623358f3a01799e8ed08669a36f7b1d84aef6c90694f99e8f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x3_rM_splitK3_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "64e7d614da0fbe4b59fe6503892213820826679c5fc69b0bf45e137f8dc063c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s4_et128x16_m128x16x64_c1x1x4_rM_splitK4_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 768, "6fa3cab93cb1309edaaf5231425e9fb387beea83eae1c17b89f90de67510cfaa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "d329dc6fbb273c7b658027f35235fbd932de940310f9c5ed76f140531f77d3c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "974688e8902dc5152ffbf24d0f31017c339a5921b8df7ce61a3c2f245ac014ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s3_et128x32_m128x32x64_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "b27054645d7b320ee947b16ac937f169102413c09e12edd62e7517efa781d001", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "a3d593b3b261c0614c1711b27963b651e2969474fbe5beda74682a32260f4cba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "d93dc53a01c50a1a06789a2c86be51c0e8303a885741f1c01f541c53865b2c18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "49873fb28da8861d9adbaa0efdefb467dd80c458d6e837c84b2ad6baa989e9a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "4c2344db26abea812a7cd73be8f571d2168641e003c9a1efd11ac582a383b94e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "9ca9ac82f7b904c8c1f55eefa97a8f1c9a2d52f4c5bf0fd266e09642f3acf698", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "8c43dda1120561a70255b2a6a6b62c2a09116ada3b63e81fc64d969e0812ecd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "fbbc633f6b49a1327520be342b37a5ff4075b113a90a511bb0d692dd53799463", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "248f4dbdeef8780a9c75c6c920e93faa0e0573d5ae6e9be8c1cbdf08b672bb5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "8cbd2467f1e786c947f89785d08fb00f7e2d566320035791567e1c8345e47b07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "daa3f45c387131e682f115d5296678fdefa354f56e89e038058789bc4f56e2b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "11458f986d0963ba7bcf530cfbd4028028aa7165935e95a002a7650b99d6fd73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 122520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x192x256u2_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "e32a053dd2f0ff40e91a72623342d987b3768a49fb733ee4827dffedf96f7eca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "ddf1cb8348d02402af4a063239b06a0eb24a06df8682f4304a78eee1dc922a17", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "e6ab3308dc22aff386371d01ae3c36fefb4880c8c46a2e124510c356cd080eb0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "73600c81b88125253a4fd4406cb7c9423df5a2892f1d636d2d59d50aa7b899da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "8682baf58049318d5feddc68a16d9172b0cf0451e9d7cfcb745296f98cea6fd1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "17170fc2c3c841a38d26ce16adb9f7177444cd14720b6dfd7b1a3b6ea3d7ee7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 122520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x192x256_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "c74bc7e503a9f1856a40add192c7880878af6cf34baaffbacb7063e3a6fe9e96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "66857e09671fabbc26eadea6d9d3c5a77a91beb1a2d1596779459b31bda58178", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "4e2be56778148db22885b399e267bf6cb6c9323fbd2962014b4f836b4fe31c8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "04eff1ae3221b551a95b7be60208fbbffd958e67794a10903788c8f1cb191fe0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 122520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x192x256u2_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "1f77f255a3877da87f06c27883287240c44aced51c72a8df6b07ea4a6136bd02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 122520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x192x256_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "cdc09cc070f49aa2ba63f824e1783f521bef5fbc3d27618c8f0aa97ff99cb311", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "cedc72cc6ca164557e3718cbd03ebe1484d1b1754f29d3b18c3f1772d4dc115b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "380240e34085754698c253ab8012cde21e99ae4c2bacd00deb0311b041366869", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 896, "7d6fb9a437e64db7f75cc9f54e84186810fd59fe99f589024e59d252730c13cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "1d7100fa0f2e7f2dd30e765315847e761af725db7e638d1769e2d8b1615e206f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 896, "ba4085f3238d4cee0570c6eeff60756d7ded09ab12b62e3ab8c2dbb3fcbba47a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 896, "06c3811da2b3b4f9f647431a834b9e0d88244cced94bc1f8f594b508be0d1161", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 896, "f9b97fcdfc8436aef406cb37e84bd53a0ade45c5cea41491753d4be7f15676f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 896, "24dd12fde2a24c8792153a380af402fa0c86fe0b3e282c80ef64af6b48a36644", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 896, "8af26ca32dbf64353f9f63d8a4edeaaa61bfcd920961e221179a439b6a5eaab6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 896, "be16b54b646663c43d42395cbbf32a156564da0f9dec0f75e201349851dbd7e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 896, "d147415902622c5582f45256cf9d7faff05a8e9237dfbaa8d31e610a3a307102", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 896, "4c87a15ba174a45b2fa67999a497573022910898a0c8c45f1d5eda1206751c16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 896, "1f95e03166748599b40ae6b7c45523b0d1ee6d753c2f86764f4ed4c2f33e08d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "b22d9fd25118df49aa4d29ab8c5e77a399aca77f2369285b13e4a27bb2cc27fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "f5f3ebe1e2554153f4db3cc4fba76aaecd646dc9e0d2133c3b410bd52d1fc9a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 122520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x192x256u2_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 896, "ca3059e92bd1d2333b59ded52657fc6bdf6b619bd182ba5ca5d86ace7b45671c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 896, "3dcaf5314eab3c3029e9096d6a90adf64e7099329dfc1edb4202cef0d20d705b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "830037962b6cea39393537c8d9af678732ceaf2c0b92385236de18173b0fead4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "ed21a4e4bec1d8ab1da84093c66a4da49ed43976030887bf212a5cf4d53c2ead", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "ecd3b3801528b190e4242fe9180cccaaa2e014b7424d372bfc96b2328d3f8e5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 896, "231c0db41dedc790bed5e90710e4ec0eb3bff949b84cf59c60d170bc01e4abe2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "ae61ef27d706c24af5aa53070c295ae39e20bbe79dd292ea1ad9317e51c204e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "fc8edd171adab5c2ea0550117a8f7dd56131ba498242d3ef5fc06b6b2e0c1b45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "ab522417052b1814c2ff7d00924b13a687d247bdfd0d54acea21bafd8b692fde", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "ba858334976caf25ae9f830e35f351a3103ed1276b33b656586cf1a3bfaec002", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "ac6742d31515e42ba3eff581c6579a1b976ab8701c32d5988b7dc31340d5df8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "ee192693bafcd8e35524093b9d2ce38816223ac5bc5f1b77e68654583c1c12b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 122520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x192x256_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_siTuGlu_dynB_sm107a", 896, "b40ac5270bede1502fcb61b253a76a7306a69b22809dc4ae328f5b7ef5f89bd1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 122520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x192x256u2_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "7246fe7ec2d1896d4df8c4e89eef4e492603332ebfdc5a762602e087d6170c3f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 122520, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x192x256_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "25746e9f479b0d92eed2720b51a41f88d2d025a1a9dad6cf6787d1c34bfe9062", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "aad13b0a501942d8315998c35f3ce3dd1031ed5b0a93cb64847b00307fbd0bf4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202392, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "d1fadc669c7f0b8277aec7d4a56b25fe91a7df9bc151dd1be94b89bf7b9f11aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 169880, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "745ff1c302c768cd515be3967a1b05730231caa062526cb2c8a5b19b0cd180b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199576, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "311e4ac30dfa1131d0a2cb15a1a86216086a2e71c4802acc9c510e7f51a32b6f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 179856, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "38b9ff1b13ad725f9e4838f8ca19bcc7f03e72313e9f5154c292c856a1dc556d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165776, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "ec953e7404c1270d9ca17ba86656d40670fc881c134eec1c30643e939844923f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135832, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x192x256u2_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "93ea92905cf89c988b7d20eb552e11eff2d0d4e7555ec63bcb81cd5b3557cb31", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202392, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "0691c829349eb4ca5a272c1bb27665f58f703e7ae49d09ef477ff23f087ae0e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 169880, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "2d078a7c469856a8e491e64eb15f0ee49d888bc732c1cb7bc543b475beac0360", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209808, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "4d1800e3f43b870823e11c3a8804347116edf10dcdd0092e0f061beddef70d11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165776, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "e328dc7004b118db7bab94e02fd62463030be82aeb154acfc6d00b3a83bf029d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 179856, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "f06ea83071771b84070a254bad6427259ace5f55f0d64acb03ca135b49420bc6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199576, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "53accd435c7bb1d3898e0ed96ef516b8c47c3cdf6c3d4b3389753ab94b453687", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202392, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "4f10ce46407d1643cc8a5fff14915bc62244c8fb249e181ddc16fc56f6c15d48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 169880, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "c78a8c240b84dcb05e6d1d423edefc21f34d232b69056406ffd85c8a1037f130", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209808, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "2d72575cb3f6660d2d1b3df89b7417729f46086520ff5e53f832adedbd0f4edd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165776, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "629f06dde2f317a57765a0670983d88b9613e9edfa8a1d82aaf5b092346f43a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135832, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x192x256_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "5c4b45bd710333fbe787d549cb7bcededf150d630b8050170777543e81fbc3a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202392, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "931bfe3d6dc0a9271911264074dda0456ab4d75fafd09327f999f58a20595f3f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199576, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "50043d187d37cdf4e1743f84cbc62ad89dda7387f5c34302f0ca456b7ebfd017", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 179856, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "a644856e95caa61ce04a08345971b95db954fa64b8f5f439b6e2b17582be68e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209808, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "35c6a6c47e353e98d58034c8e3547f5c638ab38a83b8039c14e03bc952ebb465", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 169880, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "a726e7d66421e86778b49fc270f6962c4bdb161db3d1f94639e7710b1dd39f3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 179856, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "032277d8719ad91a5c249503198ab72ab82b464aaa3564e4669cbbe832845ebc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165776, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "461bc2e5bc4acb47cc003995c72af1b06a9beb6f4cfe7d1e67f6110966c0bbe9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199576, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "df4acbe525a5f6e9d956274f35cc57e20f5116d956ae8d1b9ffee33635978624", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135832, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x192x256u2_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "1c91bfb262d051dbc41a0515acb863408494fe24446705557259b81b4422a087", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209808, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "a6f07a30b84657c942e08c442709ceb759059d44285bb5231357e4683d06f703", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x512u2_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "3581c3e05a51793f3dd390034aaa4e4bd4933b6b3e22d31d767859d0d9add6a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135832, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x192x256_s3_et128x64_m256x192x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 896, "4687470ba5681fa2d229d58e27dccb7000a3d3b2bb7814ab449cd290ab93ecc6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "9c8c674be927ec7f93e257e4dea0ad797c2729a2cacb4a86c855604b859071de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "4d4b56824be11e51353076e4258c4e60de67d31475928ba1271395a2d9935a34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "bb3ba8f39da25a7ae49fba30893bde5c23e142968d947311999b930aeb44db44", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x256u2_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "dbc1c4ca2894e4f065d6a0652555c0c2e16476a7b4524f806fc8c3eb3c3e7e08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x256u2_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "9a5b596505592ef2469bfb41556c24deb3d5534163533c30bc299df23f369d09", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x512_s4_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "1216703ef42a4745551c8a6b46e42472f4e45fcc86d718d94dd9f5f4f239f7cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157592, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "5be07c81b4c23e09f02bdfe02682c8cc98ddd2d991951ad363fc69b6e67ba5e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173712, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "202f8f440f1451833963fa9aaa175eb47c391aaf540428392381784a79a72ab7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153488, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "7c15cac52b67ebf427129d83723ad29bda39ee4f92f6d3b17ae8d6c219daefb5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x256_s6_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "3d22034877b00f82580b6bfe3b5fa2b8479f584d3e57bc87d13705669136230d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x256_s5_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "a23a617dc600ceeb2ff6cfcb9da99dc950e2f3f370b00468d3ba374a4fce48ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "73fd780d37bf4ccb19d9fef558283de1d61582a9b05ce08fe25e99e634c1d10c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189336, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "40188ca6cb94ac39d4fc27db7091c2b1bc1a1638116f36da73ed6dd4f1a683da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205456, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "f4c6c2edc4fbe1a85984fbbfbe063ffe4ad2aecc11f71c36cef2c7427c1a2475", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185232, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "c7313aea37294725505c9581b1f259e6abca1a170abf22a873361b5d31493629", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "d87f08475a524aec85da4eec7a3ca8a8c510510ad829853e44b81371b07c17b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189336, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "27dc02e13c41bdb9c4c0de45c090f990ea9ae50c2d60c1755c0d69de057482cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185232, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "b0f3efb0b38f17a8e345d01aad249b99dcba62a903078166e96c064b86aeaf97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205456, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "4e3cdd7e710f8b795880344575b3d57c9a2d3741a6d016fc2afae6a7ca5496e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "66df54e7ab8a0171664da698fc11c89fd37c4bb61cb5afac22c67e21d19d32b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189336, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "2a24a4057bc821971b381034996498191aa16b73dbe5b96c815251f81f4579e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185232, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "f882d9880c08bc2b7416525ff80c148a8e0f7366cac6469dca468b0cd82efdec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205456, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "b758758e5c278adeadd23e8c21a1a3034b44487968995f9a553ef523796810ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "fc4a09d406d1927fddb210bdce398c2add026487a8f0129d5994e0f98437be0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189336, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "359344ef229f29922ac773b88332e18027f51b1af2e75292ba597683f8e9cb0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205456, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "ad9f40aa6db96288e6a8b16b91e77d9a5e3a8dfc0d7e2e3be51cb625f6739413", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185232, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "2d6e620c84e6fa87855176758aefc75aa28f48ba0f0ad2843457aef8567df550", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "0d01dc166fd9c0c16a43630fb78ffdbca0f5ec6b033f529a2b8451ff6292573c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "a164904d47cb67eabf2333301ddda8e63d87d61845da4950f08dc3536f195c25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "0a569ffaa7549e75ad555eb05e946009e39ae0f674b13c4c5195afe488c76b42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188056, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "bc759b89b0fd8ca78ffaa712edc3067da7550ded72427a301e9a585104235b0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "c3900aa545067495df123f91a14a91d6f3d7f884ef78c62dfc6b749c70b85e09", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "b97c827e6b94bea5017cd84398fd684360711c0d9c377326fe06cb013974556b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256u2_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "dbf466669a21bcdc0a20500fa42d6d844d9843330c1e65cd5f704070696ff21a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202648, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256u2_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "50d8eaf2416ddec1f02dd19b36560ce28da54d61edb4ea860156dcc52068dcbc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196248, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "d789d9c951307f45b7b0553dc3742a743de730afdc06fe402bea1b8b6132a215", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190096, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x256_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_geGlu_dynB_sm107a", 896, "f808941a4c32ff51132661f4ffd203afffdf391cfbe0f20620a8566e0e289693", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 212624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "0e5356b1f6ba7d95a69cf654e4db741f6cd612de82a63fa3a2568cfb64913a39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "cfb53ce1f358bbfa09f15a57a56bf7155898acadd2cae7f744e0ea6aa88b3814", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 198544, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256u2_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "0fec3a1afcf702cbefca1144ba2140922c4e4c2aee51b7e30c8aeed7a4f4c5a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188056, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x32_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "392e86865243271820586c654d0323e7359158cc2dd4e9af5ed43e12f5868d8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202648, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256_s6_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "1bb925bd14305f4fd390544277f7eb66b3205925723be1056b8beea6e566d927", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 198544, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x256_s5_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "69f6279a046c856b51e9c01a33bab60ad7d328f5f8e96b0d73c05c133fcbd8a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 212624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x32_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "40ab89e04fc8b4e32a9cb108ab94ba30cfb3c066f91751347dad3692f810ee55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197264, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "d8e5baa9470d2fbde297c878eeada06789bab67372bf5bdcd261551331aa58dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "2b68682583d7951487dab6df5d8c8f8b6cfc9e4626e994a5dbbc08bcd5b81c50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "f55189cb2e0d00069f03b6a8beec187c1c241221c5aa3a834e042a4b02fd211e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "e255c013159128d647c7513d09d18b2e8783861e7be92b9bb3819565db7b864a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "3606445d9b9228cad2e94fc89192d16319a1b3304ec3025ad5fc0d312482b3b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "2da920229c33ae693fd462922e8a9b600f7e387f39af97f982d3829d0119c038", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "83e4983f21d20d25bf8c3381979c200633ad073a475f87f3233180f11442ca90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "4e5c49603e13697891d312a1df30c13c9e00f6ec32a8011f35e09c9a8e18435e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "671b7b3fae0c721a117b75eea3ea38bca704d1c21b7aa9dbc7ad0853043e844b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm107a", 384, "6cdb437400df12c3df6e82c51510de3dcc9e9bc888eb22536a09ce994a7fb547", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "07d61c8240b0d0430eef283c39a0c194a2f6b72521423738b6a4ac4f732cb3a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm107a", 384, "0355acada3607ad2682dc3e620beaf0250a72cf0342abbbbe09a2553bfa90024", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203416, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s4_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "7684979a82c51fc0de600505761038bc2144ec371be6f88ceed9c2ee421443e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197264, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s3_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 896, "2296ae4c658aac6f1484626b17d8da501dbc9f6e82e86d215fc0a4e12125cc08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm107a", 384, "4c6ef56db9c9a3d7b4f94c7d8ec078d41fb5723aeddd818315a57fde30bb996c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm107a", 384, "efd8211ec3a6754a19ef922e9e61a717a393159edd505ab119335a68d73c0fbf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "19697e6e62593c94d3a9036bf6239fdf3c48a3d02522cc15b2abb21d7b7c51e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "a02724e66c66d0ea51e1deaff01a9493203783b4af76e137ffdea66daa7b8ef7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162448, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "f4859155a8a17e2f7719e8350914bac3703c0742af8662171619d839e60b6d36", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "40a9aaa7d1c392798f7aaf9b19d815d522c91d77542d9b04f632ff88505647d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162448, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "03bdb7ddad04a8338435dad0ded88fdb701e4ff72afa16054537205361bbe9f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "551fb0f36e26651b3e3d64870488565c7a86fb61a0c2c50026088bb4257a9ae4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "c54db339a41122e8e9e30046826aa85de62ef7f58e904a7954a00c93e3750834", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "fa2918001b9d08be7b84c8476b48bf7240a69b3139786379705f98726e31802c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "20b9945a265ccfd3602bbb2831ee155e7e6efc090dc51157c3fe495baee874f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "7f3ed0de40b6fda973bbd55c029e8a385c7ebf67608f8262341b07326ebb55a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "4bf18108dc6102e99d7baedda769b9c9f7a25c91c8510ff0eb6292a627cc0234", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "bdaeb36a30d8c9378f8f06e9cb9c003a5d4d09628666be623dca5778518d4da5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "8680ef9fd7144a272a13c1f74f3828c8561d9c13f8b555d9637927a9a7191916", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "182b1b2676191e6b2dbbf08abdc936faaa876433606dccae630720749a82f215", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "75da2e46514c2593ee22d5358109ead2e507d7f03f54de09ee92a0717bf09611", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "bcada5014271462b05e1eab8224d6c7ee73a97ddc35a49df47b5f9b66dc7c306", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "3376e6ffc419bab30b9c2f6c0dbb89e1e4a18294703b0c74444a9fe62abeee45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "2dd72dae70f0bbe53abf97ed018dba0493036a30fb89bcd2f82d498a98999dc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "ba1aff7e0d5dba8b7271f0f8eb9eff1a2720ce3dff874b338937fea0d847ef41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm107a", 384, "7f727c9d1f78523984177ea69e22793a2853e6447cdbf6d5e69f719525851ab5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm107a", 384, "271caaad6904ecdbb88a9f69736026f41522d291918ef211c187d10bb92df91e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm107a", 384, "fb2c93ba98dddc4cbacd38e5ceaaffbf57df2f851068cd68ff75c10c04183e56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_tma_ldgSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "0ee40401857e04d7e431a0135f8e10f2ff9856ae153998bb534d785992f69212", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_tma_ldgSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "db122bd06eb9a8402626a053972b2015d0c94e1e15ce8a1e0c1f32651295252c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm107a", 384, "fe6db45b9d48e1e1cfc9ab33e704e50721e62f43afcc276a031cda6f8f9ef965", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_tma_ldgSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "d9ab220df5ca16d0528fc571b3546b8e97d4cbcf70ff6db8bd02bd58d2fa6c91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_tma_ldgSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "bd199f6a60231b59c5411032f38e5297c2d3f66d594fc0f9c8832efd4d121762", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162448, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "a8d3f889be9a1f952853c0eaf51e8093c945024259c0b7029df0f995435ecd2f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162448, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "67820179223fcb6fc521e045035945d0f785401edebda4bdaf6f3e7a357ce832", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "5513e2cfe6606733d984d81fcf813b8975f94814be976bd1976d2150bcef6b57", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "170812bfe3042401eb0f78f92fa173a73aec570503122d1d1a76105f8342888c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "469c85ec77173df66f19648a66e3b496cec3272fb2a55bce2d2a5c607e5fdbf0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156304, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "0db84514b214af5f3b211fada8b9a4b9f1761c5b8b20a8f4c39895be2b06407a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512u2_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "0b3adeef575b027918ba015e8d6dee4ce5933c81d5a2988b293df75af0182b8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x128x512_s3x3x3x3x1x3_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "6d87585046c6c5db0c75bb5848d1fe3685e3dff0b3c51eb2d617b119d5985ff0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "fc6a1cbd246d9cc47c885ddd8330ac14182f3aa8a6f9261cea896fec865b8ae0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "eb64fa506d637279948f2f3ba76aa13d02d17be2871f14719115c222be2279c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "c635740706a023effe57b9c0dc6a1b79c0cce15e7eef7ab5fff5c869baf37161", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "cc8b2e9a30ad054a336f6b26fdf2deaf90c58b71c2b92e8871200f9f704770c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "afdfa0b433369c2a4b36f600f0765b9d587cad13d0c86fc15433a30ccce0ce25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "7c2f6b20e53adcd6ead20fb7a38e264d28cb0518e7eeeeeb501d4f6ea8682717", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "2bb3a3fadd85c9bf00931db393b2af7ecd68de7eeed01aa2a1fad245728473c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "c5861a24ab02569f2a6184878cdae1c06605b02479652b46dabaa1644fb60de3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "9dd9dd9dd494021d7c63e2a4e5f22622ff3cf75033a124514db1ad9b47ca836f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "2863878f7319e76420cff3cf5c661661f488b06006466721862b937dd404296d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "0b270e367194d83a15d8ea2588028697430af854e272cd834364bb335e1a82cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "9ea10dd8e03d8413cbaaaa605ca472150f29c8768e404c1eebf8f68aaca4adc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 182928, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "780d1a510b715d45f319a7e24dcb3f727492ad565277c8ce63847e27820a440c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 182928, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 640, "8919eb68c950ba564a24c22c037b4e5fad4b1a9f8df24657bac687881beaa75d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172696, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "8d0e6ea3b87775cc896ebc3fdb24dee0b1223d4daceab6e2584a2a4a8945eea2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172696, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 640, "fe0276690eba8b0345cba899bf8af71de437073bfd4afed92c042060b69ea396", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "61fe64142a6dd0f9e19b239172842a91e8c7790013ecc211c62c322d3acd65d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172696, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "c778d76fb59108297825dfc1d0761ce6edf9081f792316c9687b5a56ceea3ba6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "95e9a1dd100ee688405ea1f4e9a5340868a36f2537f5925864b7161036a7aaba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "3d61f78932e3291d6db276d466fe1dea4039f2a92def72f9c404be0b4d999d10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 182928, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "84d8ec0750d619e28e7e670a8acb4621fef59cdabdbcf537f3293050808696df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172696, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 640, "3ead9441dabc1e0b9f632827916548bfc9018478b2fbd233a0335b8a631dec6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "0d48f3f4b7566743d4220ef24c63e02c62535377fe3596855a1f608d1bb1da37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 182928, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 640, "c707c36376ce852ade7c4ea415f8c271c2a08cd7c1d5beb428c96e3b66f4baea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "adf3e6be0025adc0516d6eaee8eba659bd04d9d3d8f74ead66f8653ad20417e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "06a4e03bb57ed5576f8a309adeca67f59954091785c9d1cfd3b1af6803083aff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "664c0f772073bb26872de0bf58e5e07c83b771a747b1d93ac703bf58e30c6936", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220816, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "2c8fbb84e1b2c1fb3383000ab511855d46b47b1894932c607a5628c849f4676e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "79940ef67fb42ce86450f6112d66455a09dc40412f3c77a7e4aac2171c07ed5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "e2e15c7b9c2f690d0b9845991bfc431660fb856f503921f6145cdbe1dd6285ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188056, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x128x512u2_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "ae5f330fdd3929203d2850247d7493a9f5f0369a6c8d8300647af5b956745f12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x512_s2x2x2x2x1x2_et128x32_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "874f559c90deaf17ed31476e46234f44ed58f6a7d8859b77297bb6eafa32dbda", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 208528, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "963cab900c948400d640f33f72196a7cef4976fe5ae151b7e370f08508d92520", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 198296, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "8e6e588dd889cb533aed8c77ee01844d4f9e1f170b903d9bee24229dc08f435e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 198296, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "798dd931e2a7d8c8e88f4ad3cece7de647dd57894ad93aafd9e0ecf89d537fac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 208528, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "94e47c7539b3c1a5acba30340e07f3553ecc6a8c32664954aa93d078261379ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x256x256_s4x4x4x4x1x4_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "276135248f902a9ba432980465eb0cc5f3ef3a0aef41419adab6d5d7c2c4704b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183952, "bmm_E2m1xFp32_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_Cb16_t128x256x256_s3x3x3x3x1x3_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "95e9dd5e32666273dbf4b855ad1d9380b05a72f97f1077dfe7a20864b51e38cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1056776)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(0)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "bfedbd74410e92a5c26204c3649520cb860521441ab1af4645645b46b9ebd6a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "3a0b11a2a2f5f25d2a5d99e6ef86e6f285babe1f4ca99cd3a4a54f9a27f54101", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185752, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s1_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "1cd7edfc2ae7c53cc50738187f5e5a51e023222e66cab580fede48ab7a5f6d19", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1c3be550047c9ae87a55fd1e74e032989ad8015b44c2335f67e65a17c2a7a442", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124688, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "5b6a1d8de7010fc7a1c61a232c6985630fc8b98632bfc3df382d6d08a982cc46", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b08f4ea3dc86cb181b6df8a7749db4a81aa2bbe32c589695918e17acecf1a1cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124688, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "3bda209685a09c493818bbc257587ba2a7554ed91fb48d5ff929ec5c399d703d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512u2_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "274469fdfad0ee43dfd63f018351d4c46f413161de7b64d60932402d878422c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s1_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "b9963b5dcb85cdbeb6e4bdd75d9c8f829118e99164b7aeca2b630ac4a3fb164e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "28d7f51cb6e1806e2545ba6aa2f7050470b674265c0cadf65644f6b701b8b2a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6beda6f53d0ea992fa1b34e23d34d35d6fa564b6720bb231382df9564a3b3e2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256u2_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f4b92a9620fbaf0522e99cb33d4b8e3c6f44c5ac08daee8331f0641210af6f93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7432b4a3a95187901491f875c7abda49f1d098b9e70cc7afab9b2506b0b905b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s5x5x5x5x5x1_et128x64_m256x256x96_c2x1x1_rN_TN_schPdx3_relu2_fCp_tmOv_bM_ldgsts_ldgstsSf_rgTma_clmp_laW4_lsfaW4_dynB_sm107a", 512, "26f2042002e445a198a2694ee24d2a22c8d2162373dc30d34bda38bed8a87ad3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 120
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(0)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 4
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 4
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185752, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s1_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 640, "617553ff3529685ddfa2e58bd716dd1c849be5403b446605ed02b4dc25b0001a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256u2_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "9df5356be0a0c8ce4ba074b1d2050a2a87b47d83cc9029987d5023c0fe2731dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 169616, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x3x1_et128x64_m128x256x96_c1x1x1_rN_TN_schPdx3_relu2_fCp_tmOv_bM_ldgsts_ldgstsSf_rgTma_clmp_laW4_lsfaW4_dynB_sm107a", 512, "cb3b6ccd677a50f30e6a22985e2c3a1549ff4799869b3f4560ca022875fdd7a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 120
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(0)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 4
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 4
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s5x5x5x5x5x1_et128x64_m256x256x96_c2x1x1_rN_TN_schPdx3_fCp_tmOv_bM_ldgsts_ldgstsSf_rgTma_clmp_geGlu_laW4_lsfaW4_dynB_sm107a", 512, "f4c839136a2f22ec79eabaf93e08568a7bcc297b2ace516115cd158228a4f591", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 120
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(0)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 4
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 4
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s5x5x5x5x5x1_et128x64_m256x256x96_c2x1x1_rN_TN_schPdx3_fCp_tmOv_bM_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_laW4_lsfaW4_dynB_sm107a", 512, "ba7d460019979194eb1ef37eacea0a26ae201acbff12b6e1f764384f1785195e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 120
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(0)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 4
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 4
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211352, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s1_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "05e4a79bf923b0d3d7fb6b7b3728b20400fd003e5642f3da570fcb8c4c194450", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 169616, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x3x1_et128x64_m128x256x96_c1x1x1_rN_TN_schPdx3_fCp_tmOv_bM_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_laW4_lsfaW4_dynB_sm107a", 512, "eb804e448a6106453973db1ca93fcd81d4f5f3a4795588318d57fac01e7d7d5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 120
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(0)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 4
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 4
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 169616, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x3x1_et128x64_m128x256x96_c1x1x1_rN_TN_schPdx3_fCp_tmOv_bM_ldgsts_ldgstsSf_rgTma_clmp_geGlu_laW4_lsfaW4_dynB_sm107a", 512, "f75e1948e43d2a2a18244b74ae9795e3edd9b3544a6546ade446f67106f7191a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 120
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(0)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 4
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 4
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 202128, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s1_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 640, "69c6fa732aef22a150397ab3acfa535b385112e3d3ce2eff897994bf94f75828", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195224, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s5x5x5x5x5x1_et128x64_m256x256x96_c2x1x1_rN_TN_schPdx3_fCp_tmOv_bM_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_laW4_lsfaW4_dynB_sm107a", 512, "b95c10aeaeb2cdb00a9e11cd253146899383264c5fc02086bb322ff80ef819eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 120
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(0)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 4
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 4
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 169616, "bmm_E2m1_E2m1E2m1_Fp32_Ab16_Bb16_Cb16_t128x256x256_s3x3x3x3x3x1_et128x64_m128x256x96_c1x1x1_rN_TN_schPdx3_fCp_tmOv_bM_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_laW4_lsfaW4_dynB_sm107a", 512, "115d9f607df56b02c08797ed4f18ed3a5d81143174d2645f5d4ef8c8ce83db51", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1050629)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 120
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ 16
, /* mSfLayoutA */ trtllm::gen::SfLayout(0)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 4
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 4
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b3ee069186b196ad22ddace31298483747109a0fed4e67e7e8861db0bc6e6aa3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227728, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s1_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_eW8_fCp_tmOv_bN_tma_ldgSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "a8bcd103a87fdad6538c5c9c4173a2a4a1adb906a672f9cd9b1b0a9c16e298e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 32
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 40
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(3)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8824cb188b699be660dcb6d934737d092b1634f857763909a3ab8c7ca3c788eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "723585dd66b167d7f80b61c925dee2117cfb26b169bed275ee00250eaac3cb6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512u2_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cdd216b991a481191e9ba8b297b0c2be0feda1ba6ff3e9b6178507e7b591f314", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b90446c778dc3156aa9c86eda027ae6c6c43a5680177d231f850cb2fd536234f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "25c33a4070432ebe77b3520edfe70b908f54fe20b1bb838e7f935d001fdafc05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133904, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b67b83a013e689a1a4f2b1068b725d0e58a2ed7d573493e4992b01e63a46f426", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133904, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256u2_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "5238feb89cdb529624ce36d3fe3314981254e78759c5c72ecd913b6ee8069c13", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d5f497ebcfaeb83548f7b4ce751121980ce50117ea86dda62831f6df8b105bc6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e7578d928d1ad62dfee2a147f290975e9d09e7e874527ea5300e356fc2232cc7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "65cd0909bf239a873a9a73738127a0ec4d581408603ddd48b1d5876c474e1ba8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512u2_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "4b16f7ee162be404429fa0c48e74527f8f7b035ec90223ef253f4c065118ba1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "790243f7dedb2d7038303f7cfcee6ec68bc7d5b780a6c1a39635c435bd95ee6e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256u2_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a3ff31ba54201bc8c38c735fb874ffbcc1faa89858ea15ceae12359b7a7e42cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d7f30c741c67ec146b0e29a5ca529aa22fbf4d0d5097e9d37710802ef43a6144", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256u2_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b7261cf7ad78a444c2df4a205ac0422f9037dbae7adba8468c86dc1cf21e167c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181784, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "31290b9d26c64e88ce0347ffb73269758ab4afd2924b5c434a1701bce6cc3ac6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "220f0e6aeaf3ba9517a5bcb5855d5dedb7fffe70ceac800717e7f229d40d66e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223768, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cf529af55dd34fd6214d30637bb71fee1f257d37f1c57af80616774dea386288", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a4fe1209171e61cfd0500f200fd1f53dbe84a6ae6a26aa661e341905064f3ced", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0ac171ed506a8fbb726c2f83b899222af969b336d7e198435105677950970823", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d1fd2cd54d5b1ed13cc50028b7e9e0cf91125d6fafccf54d6e47dceec7d6167d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151312, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256u2_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2409dfa41730a80d38a1e0228bc4a19585d95a7c0e625cdd407830730e88b655", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cf2d418502ef0a1f03cb410c7fd3c9d688befe85578b3d53137dccec24feadd1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221200, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a044cc0e6749b7ff0e70d6ebd50763bfa78496c2c706e59db1892cc4feb52618", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "93f4b961c01116684ce522d76663c6c2a85ea7738e2aef10ed102af5431b0647", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f9b1876884ae348ca8dfb56d99706cf302e00792fe1afd507f4e3f651f7c6492", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151440, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256u2_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1096460c09c6374f601109cd631841be439bef1bc7ebcd35005fb599b72091da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8a49843af6e1438eadfd52eaf35367d43aadf7a9e04a41b084617c20efd7b5f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512u2_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1253c3add230d705bad158e08ebf6f472299672a3526f9e1772a89ccb4904aa6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222352, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ec117d8c14e8b10341853e691182384f9a9c8444d95ed491c5096bf28426b7c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187920, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "fa63065cdace132470561c6ee2f21a68df23ffbc23108b7de89cf988ca0fbc51", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221328, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256u2_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "47affa40318005197f2e73f2c63a3beca3c94c2d2335b3e269c5513ba4b8f8ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187920, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512u2_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8709280b51b2d6b5967a3ee895834d72146955ca81c3d3dc62d21e9a35ac1e7b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188048, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2af6ecfeedb0fb2a148c6239fd6b02e928c25153751af228c385fae3ced18c0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188048, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512u2_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "56a13db2caa0c91c2cfc46971a7cb2cba01e6600012590f97e70e9e9b189e185", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1c65e862625cd6f22deb7b69cde85e0d7792afb7740f2b65e87ff8ae5a87bf52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203280, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "70383cf9c44b484e75f22a7a231cf07eaaa4ded2994ebf4bda0b756a99827cb4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124688, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c6828595a838090c4df754cf1f510b65f39061527f93e7f3bded4440b7731361", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "147aef39402aa22c9ec2e8e3fbfad6d5eab49c5b1e5ec57a80a883424ea2125d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124688, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d7b7edb159d836247084a4172bce0397fcdab19eb5e9c2ff0f087030666de966", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203664, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "628dc5442cbdf30e202ae9d446daa7419530e16b4ded7a84bdec60605938bf33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e9a59ccf9810070ff2b69e5c769b326bd9e462ea03faf0dbc3e112469ee30e27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ffcb954a735dd910312412bb04c8bf48f272a08c7ef74e65ac7bdc9fbfd559a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "19e8d4c495f68dfc93fa36bb964fabf16f113e59330bc69645bd188e65954902", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s6_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "06a5126696d6680234df8857a1750b9fc7d585b8dec1d94199be3735b9412d95", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7a4c59ba276769966a822bec79d12dba8239d67d90c4be05855900a96308578a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216592, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6dd4769b918aa9f0753faff0fbb04e2658cd43ffbc5608f1ef8df6a06066674c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6168ad5d1e70179b6639f6b77bdc0ced86f3e580336de2f417d310bbd19e974b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x256_s9_et128x8_m128x8x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d26771bc65116b256e293a0d09182241e5ce1eba1b80e4a2cceaa33bc2939425", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "942dd1731f6eb21140ab71d4a4b332bb78cf3bd21d1cbad9319796c85d24a9b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "36d25c3c40176a3a4c55fcd8fda93715576018fd6f668dc6669f09fd6a464f0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206360, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7ce4f02db90cbeecfed4e8f90e2163b19b02d2220ce991e0574c6b383e7d3148", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133904, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b0afe95d07a548b107e2270c76fc95d25fe5f11149257d8c1d7beb3b4b91f86d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133904, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e485a585c75def73a5a455ff3b3ed91b1884384001c82f08a422caafe3f4b6fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216976, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512_s5_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "3e389dcba5ae0554296a2b3507affd1204821202494adb550da07d76d329864d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "80df1cd5896861b2cfc82ec1f1416be9473ad7971e66a2d1fdc9cfeedcddaab4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196624, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7538c26dc542b237065691620f850f47eed56db6ac2ed8761027ece8bdeba394", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "92b83ac8747dc080cc83f10ede96971feb4f5a56638435b43132021156260cca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x512_s5_et128x16_m256x16x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f694c10ef5da771fa9001cd2437b1ade02b2233666c2b3e05bac6d1a5aa453fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "24f82faadb15efa4a3dd14e97f62b18801caa5ab65e0d300eb34f65886decc9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134032, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s6_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "804621ad8843174514c4dcae2276d2a23499ef1f8c45161917e72cbc22cf26b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "666f1396a2ef6dcfac2ebff15e3fa28d25cfa5ecbc124919e1d81fe935ea0296", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196752, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x16x256_s9_et128x16_m128x16x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c2abf7728d2b652a6ead5046f353edf64a6893a71828bd71e1637e92f3a26bc5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180760, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "43d38dccf43994ba403866df25db102b4e09555a8cae3cb5f83012401b8ea360", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181784, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "76c6de751f93fba8fd3e2b5571215dc299b9fce521680171bd32a574f03f8378", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0c7c3ae6126e81722533e1e549352e9e1b32233e230844a988385987456d7a1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180888, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512_s4_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "36733ea35af0e6afadf906008c94922d9a106c7d9ed948efdc5e2eaa4cbb2086", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223768, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "48f3408526207895ffaa10d3e794d8e1dc5be5b8270c6484087bc73d966b7d7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222744, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7c71bcfd5f765f02c000f59af4245eb9c93cab1f7c2b226f73f5ec2784cb7006", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152336, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ead7a54aa4179c84805a9c50009a5baf5bd8aeaa2d04efbd19fe955acdc5d233", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151312, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "9265141a66281aeab2679e94cbebe9e6baeb40c28f098c66107829d698dffca1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222224, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b14d4a7ba885b12fc2404d5ad230d104549b3dd58d192029fb23b441fc2f42fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221200, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "86d365beada4cb47b534f74712ec726c40868141e332565d7be5bfe10b9caabc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224152, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c1dfcbcf8eb7dbf9f39bdd50a1cf2ad6d225d0bc334bd02b83b0309d38d7c0cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x512_s5_et128x32_m256x32x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1f7c1778f01944f2ed0c7001b0697db58af6aaf62eb43e7e766c697b268d5bf8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151440, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a42736d17e228166d684638e46fb9c140e65949bae61715b2320e80c025cf683", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152464, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s6_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "fc65a32a8cffe5a148cc6a757ef2d6e3e0e43a0707300c3b568937c13eaf2bf8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187920, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "20ef6acadb6eb4bd1d31f368f3652bd8438c967a1b2fe81bc363804b338332ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221328, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a506bd944ce2687d72344d9331556b5596d6d94e7b294f6394bd315f37d316e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187920, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "4b0dba3b2ecdd0b60709867092501682944fa4c9fec3a1de861e0c7e7107f565", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222352, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x32x256_s9_et128x32_m128x32x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "519e3f25f89ff2244759a59a740030f2d553f092fcf058817486a3b863f56aea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 9
, /* mNumStagesB */ 9
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 9
, /* mNumStagesSmemSfB */ 9
, /* mNumStagesTmemSfA */ 9
, /* mNumStagesTmemSfB */ 9
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188048, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2eeae2b94c5a06b2ffb4e1a4d56dc04230ea0056dca3fd606bf3ce12b6fe4142", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188048, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s3_et128x64_m128x64x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "faa168d77a9db7ddec343594a1ecca76018e5d5fd7e2559e36f421e9b78f2205", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "5c5072ad62af1767b7bbf224ee385a9903d3876068749c2bb01e9d2b07f9782c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "01df5db965bb50f1ebf9dd0729c8ab5a5eb35506af03b82a40f4d1bccd4cd5ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213912, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x3_rM_splitK3_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "eb36782f484894f8e935b2871c7fb5a1bac5c9bbfafb6dd55d47082b3bde36dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 3
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1536
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 3
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1536
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 209816, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x2_rM_splitK2_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "44b109aea16366097f17da65fc3f60e0bf571b2a3749e7939305168f39a67aa7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "754467cc5bfe5ccefeb70ee96bce4bcd392f8bb17cf366ec6b8ef5e54ce3ec5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218008, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x8x512_s5_et128x8_m128x8x64_c1x1x4_rM_splitK4_TN_transOut_schPdx3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "fc91743eea020ec860dbe05d5e9e0c410d88d6c01a0f03ce186cc1ad0c6b11c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 4
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 4
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 210584, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512u2_s4_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "85f3d14c6ab475780402f1b42d06769af1ec344a6b408a9353a395fd0fdb0447", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 210584, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512u2_s4_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "050ed9ae1bc5209cb149c4e544764df22d4900b43f42dd3dbf7f18dd64fcd8b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x128x256u2_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "f3454321707c64bba700c0480003cea289561d3723463b8980c283d08f1cec64", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "434b08bf6121cc383ed02c53c8ae3ba3467d9d2a88917606f7e5c15f58d39ea6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187024, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x128x256u2_s4_et128x128_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "ecae14bc64101521f7f923b86e0b430c1cf5879e9f916c8f730764b4059b624d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187024, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256u2_s4_et128x128_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "00d137101ea00fd86ece90b3119415f8f133d7c624b633f67355c11add845328", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163472, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x192x256u2_s3_et128x64_m128x192x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2b2b35b0929c53f487ea0690904f2a04b29455964637e87fa322108de71e1515", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162448, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x192x256u2_s3_et128x64_m128x192x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "affad9d926842a5610eb0baf709f6f451d4fddccaabb8e0f06009e0745a1aa52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 210584, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x64x512_s4_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ec2f7f896334d74dc30758135a765b8e1ddffef370a4d6ea218904d66ff688bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 210584, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x64x512_s4_et128x64_m256x64x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1323d76fce16a72541d9f801f196c43bfb2200de05f6246b0f1ce78c0595b204", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "a0b8aa7fccbb54031511ca7328a2cd3035b922461bf44bbd07995cbb5fe9242d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211864, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x128x256_s6_et128x128_m256x128x64_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "ebb79437e32dd0334b2ea2c56aff255ecd5ceb68c9e80013d2a425b8fad91346", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187024, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x128x256_s4_et128x128_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "6828297ed7287c314d1a4b0b506eb662d6e3c9d9b38773e2b1004cd2ff6ba80f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187024, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x128x256_s4_et128x128_m128x128x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "8eb39a715c7d96bff0e3ff689d13dc624c22df3e5fa830d2fa26e3d99e2ee94d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "dc35b68a5d8d79f53d416618b719bce620ddce37e780f54ed41617e8c034b360", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "fa10cf14195e7b8ac799a320fb662db40c9d72f7b38d44a50819ca11686e62b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162448, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x192x256_s3_et128x64_m128x192x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "dbc2fe72f3ba79dc378bbd1034b2545b888ed496e99483595c648809034818b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163472, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x192x256_s3_et128x64_m128x192x64_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "00b7f8c1ac58a1bb5c5cb1935b2dd72d33b8f958a7a20d8a801b1e472d699b0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219800, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x256x256_s5x5x5x5x1x1_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm107a", 384, "19217ace9f69c1fe3e09e7415caca84578aa9ad02425348e9c36785d58a69cd3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "ff866d8a709a60dafe49f3de9ecbed8a37ee413ca6a4a1a8676f001bbabac4c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181904, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x256x256u2_s3x3x3x3x1x1_et128x64_m128x256x96_c1x1x1_rN_TN_schPdx3_fCp_tmOv_bM_rgTma_clmp_dynB_sm107a", 384, "b38da8ff300a70ade5de9c44e356866bb88413f84aad3026e570a2fcab6ba2ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191120, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x256x256_s3x3x3x3x1x1_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm107a", 384, "7e8f4f6e2e0bfd754e40baae550171667aec79c9311c90c2c4939a0f8bbdb494", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "33f9ff4a5348080e1128ca05062dd895c6a8e2cc014de303a6bd3a0cfeca87a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207512, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16_t128x256x256u2_s5x5x5x5x1x1_et128x64_m256x256x96_c2x1x1_rN_TN_schPdx3_fCp_tmOv_bM_rgTma_clmp_dynB_sm107a", 384, "6172a632b7d110ff0fae1d045d8e09d86a7f805a81df76d5ad192ebf8217da7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(1)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 32
, /* mEpilogueLdtmBits */ 32
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 96
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 192
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 0
, /* mUseCustomizedMma3xNvFp4 */ 1
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(0)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "af058c973a5318a393352b16434191751e837012a254961e2fbaf90e562d0971", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192144, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s3x3x3x3x1x1_et128x64_m128x256x64_c1x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm107a", 384, "c0f4df696b3f7ec15ce1fd32c311962d8a0e38e0664462b2c02634a4b44b6fd3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 128
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "fe0996f029776e625bfc047d37ca8572803aac426638ae5b09feaa97e8fdc766", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "00242d1834404e4461573c104f0ece5ff1c8bbf58132aeb609931dbaa69d9cbd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "1150a7da579e24e7189b34ffae4857397e84adfe1c5f2a03ddec0d1ad368f9ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220824, "bmm_Bfloat16_E2m1E2m1_Fp32_Ab16_Bb16tokFp32_t128x256x256_s5x5x5x5x1x1_et128x64_m256x256x64_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm107a", 384, "c41bf896103d2d9509c2378e8ae043e29531f8e7cacd7c0045f41d13a0da67f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826818)
, /* mDtypeB */ trtllm::gen::Dtype(17826818)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826818)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17826818)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 64
, /* mMmaKind */ trtllm::gen::MmaKind(4)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 16
, /* mSfBlockSizeB */ 16
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "9a3ddca3a8302c80d0749cdf41fc88f378db0c412e5eeb79b0e5575c922d8a8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "9448635caa38b2bd9292ea55814a1eb32f5737993887a9df0eba5954117d80be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "e16d3f437d70dcb9c1dcc7ba565377577e6e48b7064786eaac66dbc971853fbc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "7e053a1fb5863fd8f2397e0cca38bacda160ee07a992ad6294bc937f946bdf72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "9ce897f9a2b0459a7c0ef057bcaa89b142fcc933ae90de29e6b76414bd6d811c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "db28d615dae307c9416bf47a2f03e865592751052bda14f9d8182f653847f9a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "4c791df8fa88298425f32abfe68a6072c7ab0a902d71395d290a8edeeacb459e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "b7bc1a4865765f3ab3a711d5b818b6495a355eb14d31d0ab04113ea520250026", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "663a9a9cc7dea60814cca1d6a9c645fe0025c249612d51d5ec5a8c6e7e19e34b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "a32c8bfadb8b1f0a75431fc93829bc46eceec5f70ee517ad52722050a4914d88", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "7300e89f9bf71a5e07d87a8cdd2ae8ef731a2d02c979f30f5e76355b71725802", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "3defce5eb678f84d19bbbb1169efd060fb475976117bad6bbfaa52b43194f7e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "3205ac12ac9e33812c8bc634ce02d3dd475ad88a77731c77157107292f4d700f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "71a639da9d3d840438324aafa0e2a080b9eb5ba0bde8ffc25dca2da478d813f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "f909814a842f687411c3c561b5a81640ed276f3c0eb3b0e33617ee1905b480a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "88ee694f051aa0db6a5750677deae261c8da355118061d5f11f8ef1c3eebf2a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "f5b41662e527ae0ca1e1d4eccc8d902920c4cdd7205b7d8715a5cd2fd0ede6af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "c4e59a951b7288e57a47240bb41e2b787297c8f92bfeccd2f029cbbde5af5fa1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "9221403c2d2e49935882195a33c14cf9eb85c7de95f6885c76b9502a11ed36d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "d2ceba3d5eb083041fa00b7a2b3f21f791eb696cb12c723d87665a16bdd3fadf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "b75b7ba449d9abeecaba5b58db67b1f2ddd3d8fd12c15a3a98aa5f9e43185d27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "ddc4e728b572bf50df7832207256249c4dc306523dbab07daadda999170c5927", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "82ecbf1033193d110cf3d8aee097a47d8e961e0166a141c391b5ab4dd4672ae6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "20892eb7baf9bdd0601a5242848e0fb6c479f2863c07534ad5376f809c1c18dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "e7754f35c1fe95380cd8279dca26eeca5e252b1e0e10983b9013b9a394440552", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "6a952cf2a639cb3f0b4727e88a2fa57aac6eccb8367d48f9f5e5a3de4707aad1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "15108509ed7622eafe123ef3da8917f6fbccab99bc2a9e2fcc3e858c098589b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "c4c15cc3facebb5b19d0136422884bf12181d9ff87f29d45cda7da94d7dfaed6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "04d823e9eeff96f86cc221011688b2a11de3d75d7f9d233d97c72e8b35895f9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "3dbd7be50cafafd07ee566541517463ee4fd813f9fad7d6c353d306716529871", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "9046cae9951c3948c88f18b67f26e32c48fdee7f4799120be5bb180e7a75144b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "f3e02f4c53a7944e96f9ef6cceebb4e18d3da7e9e5f72bd779cee39e5118acf2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "c2cdccf853ce5456491c27eb1909ba8437aafbfebc7a2af60f178ee88087fd67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "28c8514a87d50a838f8512377cf911349b06d6c1b2c59c05f06f8f4e5e242189", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "8c6adab321ce3509fd64fb3b041d2d543c4824d2defaba9d8761cd3550fe337b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "b27eb6721b420253cf5259963699266f893568c017ba26e40ae87f60317d1181", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "6aa45140f5a0c93c5295079fc607525f075cbd3810a05707e24bbc3c241ed3b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "5150fc11b93e7aa3bfd617cbead99dee59752d019499beb2b68ca35eafd7bd25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "5e816b33647d269a99c7b2f5c0e2e8f8eebc8869ea68f82e9fd519134e74fcb8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "ad84c30ec332fe220e420d8aca090408de35f0f70ce317d6c7ded9469da9a934", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "661a81a3c94cd908bd1c8b25dade919d1aae634158c728c7d0cded262bf464d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "7d1c032ce72190508f02325bed65393350ce231f5a422c533b72719508d7a332", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "df26d7185471fb03d3d1d58aa19a10e4cd5ad5e73c5f84d4e21a1056c8bcb2b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119088, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "12369ce48f15ab294b9e3f42178e81bf43f7aa833dbecff1359b44787c2ee13c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "6c0abeed6d4af5f8122a7fefb994521fb6afbecb837063e165e6164377a25aaa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "738c582b7cf8eb165d3c269823a222ed071ad938fdd40c20c714f7e0371bf6a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "b8fd4e074de60f29cde52e56d82c96033f03dbc8c121d9d1ccd2dccf754ae5cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "03155317ecf145927f057511f3b0e06fb593862f5a4340125ec2bc80053adfed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "1e13923087cd620cbca1c3a01f85ea7730a3d6f1caae3a8b59b63ccc4f3af59d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 119216, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "e6ca5781fa02b93aff5146151bf0f8f8a23625f85bcd176064df924cdb010569", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "a6342f7c45143cd863bc5a14c52fb2538e6086e9559b2776074c0a01dcd71ede", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "26c1f911fe176a9cd2adc0ed7be1b570321742533ed81656743480b8fb6b5096", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "8892d6042c4bb3d21f5742bc40311b35713f7ff2da11ec9b5148ff123b78f23f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "e5374d1d65dd25f9b1457bf701052c5918dde066cd3f2ee23b56126f500a2972", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "8fd7338fe1bbb5d6812bef3079a73ad97dce9283f595e87e278248e56b2edfb9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "1d67bc8941a8c0c37a8336b1072343927b73d7434517ab0c74bd82874b1ac720", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133552, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "8a6f891538d6f88a702f85899b342cbe7db1b36551961f91106aba9397eab287", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "a39f4d55cd8b77a29f862486969749178bcf9862775c71d28f6571169b263f7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "ae048d32bd0b7e89cc0114824f78541f90199261e1a717dd0603c722f71f7ba4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163888, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "2ff95311388c30de9c21aedc0ebe5b236eef4de7a56f9d400d2d7b1046e7b1b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "cf6374455e90b72dc1d21e26144b6a121562ffe5724bc4a03d881742ee2836bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "a1e2a0dcaa90027dbb2cf87d1a175df25928fd5e06efabd2045e06ffde5a2734", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134576, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "f93590d0e7556172c4c069568ac10a63ed18444152e1e9c9ea361e253e1018e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "6ebb21d63543e11346be94dea7823b52eb980d2572c91a98a83fc0ba69e749b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134576, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "b0fe417378c5aeeb48a2e885885dfe208d271c4894c35afac2749b38d7a1aa7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164016, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "b7b8cd815e0892234bc6988f98dd95a652d26d969e4a6cd96e0e2fb8bb508ae6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134960, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "e41a76c05e3c83d8a4bf514a11bc884ba395ad42421e360d1617d7c6068b6199", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "b359f4c5c25d36d89c64c71eee9ba1bae22a306a7270716482cee638779e92f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134960, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "8fa29124adba5854e0db7d903ba6f333cc4529ea27098a4c0fde5b0f1bfe9524", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "9afbe49a1ae2df7d56050e28dcfa047c3eba251e1542b542e3b235689df0ed36", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "da7855c1feb87fa9c3fc855e1d997821e4c9283a6bbf145d194cda6ce26e97a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190512, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "1230c27da9732c890b0c9d255511eb63f08ff4157776f2af1253ab905682a77f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "f06e6bc0685e634d5405c51083629118101ac5041302cd90b63c363d21830a54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "51cd562bf96a1c394364bcba5fd73a4a4400751fae9794c8c2d5a4345f175ac2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "6e40b758810d61e138bd0beda04b8699c371eec7c0f2e487ad6f4cedf472053a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190640, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "4a5b8c2ee82ae28d9c9b3662cf326a507847a8039be88b816582d240f3d3d76e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108720, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "48b8b157c01a70c14c0da284beb32036d7cabcec6d20a250c25d35284f457f2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108720, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "769c91c9d91c7caee86a0f5d2a9ee61e13591ede680bc3ba652c42a8f7d1fbd2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108848, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "340b6ca9e17f4a26a8a2a926cab4c1808c9d2a26008b82675bc7437e6506196d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129200, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "1626cce3f67d861508b4518e9fd919bb66a2c138dabe904195ec4fac3d95a610", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108848, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "1f7f538c9ef23043ba0379284ed381320d47046602cdf0e762936c882324999d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129200, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "d36e7d40121c083fbb22cdb48dec81d6b6751017f4b26f4130869961c1b451b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129328, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "d33cff73dad8b9174e047fe0250d22d08a1de00c2aed9df0d9fae8ac3eb5b920", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129328, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "be5f059554a70bc4181decd38792e3eaacd2fd1f8f16c511008fe4cd59be3839", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134576, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "f0c79fa1ed6ed207771a7a2b9298b5b2b548759775ec86f43f5c1c0461b5278f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134576, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "9f20fffa6354c939b93f6f776ffcef5e1e18ae840c2f3bacdaacb518c5537715", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 171824, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "9db9f1e143cf94a1941e483d7e5cc7b4a9c6c45f5333aa8853bb50a458ced21f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 171824, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "7fb5709ffc37ce378dd1a41188b8347bbf9956920bc7b15bd42259cc897bab80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134960, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "eb57d86cd053e3413a1ad983ce15c36b52f6caae5f73de99d3ec35875037eff2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134960, "bmm_E4m3_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "401f1286297423bb664fc9950c24a12e346553682aac8bffec7a3efd67e13f32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108720, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "2cc350a0f2dbc4a86803431f7c79cfbf4e57d99da4c6a1dbc6acce466c4d0426", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108720, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "2a529606c5b9232376d55284ef0a0cfd5694fabb7989df8f5f57629f6067914d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 171952, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "b4b4010cbd6d14be99d5d6056647252339b9815ea52af53e6ee327f4cbd34b3b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 171952, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "5d9e4531175930cdfba77f3a8403a3edfd4f700f73eb562bcf970eca5bc00c4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129200, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "87c878f893f68b49e0a0d610a172b6fff5904d8ec21567d05d0c262fc2f97b6e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129200, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "5782679424c555344b4092bf1229f16c8cc5acae83c9787ab547a54855fa3540", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108848, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "05a07877be2fc7e3b7234115eb8e11002d0805e088d07df5f398c00cf3fa87ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108848, "bmm_E4m3_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "a1c96734c10c5daee42bdb0ad9f91ef0d7eb2bc7945cfdbf4616fabc54355763", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222496, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s4_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "955ed3897ac3a331a4d5f783b78bce0d01e232a5b7c93100f9c16612af4b3205", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222496, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s4_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "2a339e7e0b4c6a5361192d7f9a1f521564e545b0fb324181451b35076e36f312", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129328, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "73fae6fb06063a79cc3d338900be74e0560d1747320d0d1f9c62f0bc5744cd84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129328, "bmm_E4m3_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW2_dynB_sm107a", 512, "62ea4e7d3b4436bdbb7c84cb1c3d5efc9997c842ccc329c8578c98e7ba463e1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 2
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 171824, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "11d2f8b2ee6783588a9ccc3bf4eeadf81df49a8403b7c7fb1fbe6bcadddec5bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 171824, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "c80e01d8fe73a30086b2d1f14a5b30239e5e11fb76060ab757b0bf8845dbd81c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222880, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s4_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "734309ef9e5058081e3495f291c6f226c2d0482f5c7f3ae625a3f2b98cf8c3d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222880, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128u2_s4_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "4c1144cc83022fec556b51d8cd492fe9a9777d067e07d8978ce44c5a9e4135af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 171952, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "71367179ea1bc2039a77dd51e92386167aae8c60e85b8435b906eee33bfe5396", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 171952, "bmm_E4m3_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "c7c97b675166c419a49749559463765058cefdda1c4bbd990e1ba22e60c90148", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222496, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s4_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "043faf68beadb7bd6a9d88df94e629c74a0e041d6592c26df11d4d391b2fa778", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222496, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s4_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "a56e88ea46efbed8407ec8863925f85f466bef1dfbd62c6382d4d89f53fcebdd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "6b2c7ceb858d7a4f5ef3fb53e7bc7d2051408616a5b194bf50bd86f9f0763132", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "ec84decba6d21291300e792079ba18907991deaaecc38143e2f3df6bc3f7958a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "83501327ab9199332e030e9bddfd46bf43ffbec21442b3a9dc0093f8ef227c12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "63c4e64798ef207424c1eaf38ee917c07477946215227f12f86388b4de245c68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "dc412e0c0f9f8e8c54ed16858027e5038789a2bd217a5e2df41c8c515a722125", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "6435159c00e71685e34342ec54a5df791bd4c48d1460fa582740bd6b93504dcf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "64a1210b9e5ec8c543e3753fc0657bf4ad5bf390500533bc049e480d0230f849", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "4d218c6c6c6ffe4295fb117e9d57127a044d0e9fd650ff82a9a578c5506a5b3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "974ce0d2545a60c2753a72745833ce8e40893e9725fe2a8e95c318259249027e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "b2ad0d271ef9b71b38f50d28752ee25b187dd128bd00c46873c7fb877a219811", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "f315633f74a2c7f9294d31c07daa97f56f8bdaa83dbc8e8dc2f034048aba95c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "ced9cfbfcf63ce253ef4bfd4e3047bfb96e7d2ff4137b000f18977bada7cf2a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "7b486cb6801937963dfb5d221c42647420f65018ee992a1f3bdc6cf807b4a9b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "7cd857ef11eddf8b299d48451f534c872e9e93bb5699d412a2c04187d4ba0555", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "541db7651f2cfdb0428f6f70207cb033d51c5f60ca0d8691648923e1f7affe71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "82881de5c6623636d5f265316906d4b22c93f0f2602d296e634bd192ce505f3f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "c2f940470618af123683ec98582f66f5c95067d947e5d28ac23dc11f6c847c39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "4241b2138fb496abfb46ac47e17ae76cf50f1ebc95dc08fa3d30e3f5e5c25306", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222880, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s4_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "611e2bc5786247e3afefe7a295da80ef951dff68aae52c6f1c453ae36237d46a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "ca3dbfa0cb7a6a92fdf1c49d8304b9435dc78271414fdd67f4e13894c17f550e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "3ae563d49bb72b0e40e88495da33ec1885ab17e75a46926aedcf66dc4028f8ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222880, "bmm_E4m3_E4m3E4m3_Fp32_t128x128x128_s4_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_lbW4_dynB_sm107a", 512, "5f2ca2dd4afdef47eada9a25d45ddf073fab07960c3fbbadf84b7676cf05f147", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 80
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "c7354233b912fac28b77c21149f6fbaa55c2f2d057e879b0e1171b489c2ff54c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "d9dc463ad7caeedc75d7cd2145bb7e55d3a6002fadbd9ac1d2810b3bc4f09c2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "339e6ed2d579bb54e611174c0f618f45250fde942b1d0b737970116359250cd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "eba8e65e4e464fbe49b133ec0368a9514f9b5516a701b71d52ec2dfc66a18e5f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "45f3c77ef71280587e24659c5596f10d8ebc34cf1439ab6e2dc29c847d7455ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "30e7566ac4a859f994d328559d7961abe66ee1296ab989df6ba89cf0f4829ab8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "e75c4eb4a66b473afdf1c2e4935c72c6e57b735ce05d5c2ca42055a9a25420b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "565b9e9cd1ecdf71294ec70181584d79ba128662c86c5fb1fb39955f89cbe7bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "21508bd7167013321291c757bc6ca2550a72adce1ee510e5c8e90022d42cf721", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "ce4cacfed2e0ed5cc291cd88ef8a0dcf86e035758e34bcfff3fba65b7b8bbb1c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "06fea3ebb43108923a9a2528a97c7482f06fd90bdb278e04863e95110415e3c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "80314937c677fbc05e8d6a4817a807acb1be69c43a2499a8299002bbefb4a59e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "bcb0eb2fc211c0a3e02d6dbb62f0f37077ded98621c392372f0db35ce83d5629", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "a9e55b4d17c997ecefdf67bf75ca69d9d60dacdd8c65015e1dab64422e2aab2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "d97249c59ce8bd86c221acd00bb7fa6b604eacd05b16adbe4cb86a39333da075", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "6aff5cb9981c3426d47d015de52042984d54ea081aa86a79344dd002818f68fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "f0f2ae02cd2673e73b6fbdcf7a782743876d261dc7b031d3a3a0d11a8ca5ef3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "cceb538010a08c399225d839b4a2405b1addd6219fd86e9d850861b92e76e8d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "0ee5486689322650cbabeef57d196fa2fb951baa21f55afe59fef87ad35f17a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "91593156fafc93f673c2f563731a7e582771a43b41868d6286a90336bb28c024", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "ee58bcf05040e9edbbec346ea3c39985fb08434df084d1b15dcec8c02aaf4ac0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "e565c2d0b52e925cfe137f7377a2829e8cc2d072219cafa98fa7d0ee706664c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "4ee6c9972c4b4752a7e49fdc89148bdbba6aa720babecf01de3a7210e011f03b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "0d2a326be5fd0f509faedeb6c9eaf87aa7b675ee3e67a5c38449304b3d8a6183", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "166aaaeb17e0244452280743d06aa8df12f2619867c388c4e1840e94f7e358e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "02eef62dedf11a0213e8a3fb73a57709c83cac61941d34ff8839e9e79a8e6d84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "5b1b556cffb430057c725775bd8c43c410b0fe137991b68b6ba6934216fdb578", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "1ea29ee56ce8323f645644ee1e6ab05c1458ad2d0c59e490622925814a810dc1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 148400, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s8_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "7d85d751c8c377f66b90d743b20471957bd16623c2452fcef2f8c3033d4b18ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "1b42ce15828ce0a45221e43825da5276957dc78cf64f42d6199053878dced6fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "6d4aab2962c73e01b804ff9270777d01c73e0ace24f51784270c943b28efb70a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121136, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "6848996115826371c8121f91a477e4aa801c5d87cc1b3ad127dd332a968d203e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "5036b15484a893c460be9e150e4fc21dfc7af3f18f3ce5d12540fe0097c8c927", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "24d3622508e9356adc2c228e57c490cda8d0c0211db4cf0cbffddb463e331072", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "d51acb83a5e2f662587dbb967b9a7904c94db01aa5233120ea6b57e6d1d8f570", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "4c1845197ee8c6cf931148b5b0c2635561fbc43578522a1a587a9070fb59e873", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "721901fbb191eabd9535611662cbf175178a7d07115543d2625cbead6e501443", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121264, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s6_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "ff74956118f960a807a88b7c68af4e6b364400b9a606495e5eff15f30423dcd2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "ea71f6775970ffc06913497312b7ef8782df09ac8fe991bcf182bee3c8e05854", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137520, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "6ad7b17a9886f55611f43bb39243c6b92aa07c4d1f365eb9b9d401877b44d069", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "a4b7e5ee6d3c2de5c1f969cbcf001ce97936a5d0adc86c688e10e32ac7f23e60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "68a70f7bd65db180dffb3283f6865c08ad801496a7c51e0df559c95f981f8b01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "7ec9631cb47c526c20b3b93db196eedb255a4f519f75cdfb5f83209f294ea98c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 137648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s6_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "f57db02f2adf2ef004332b13dd4c4f12974f69448f96b7f6f85f413112e9efc5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "303dbf87081a240692fcda6f58c1d799088ae0a0fc1c77e10a527c4a8bf803a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "99d0b30aa17cd779967e8a0ec324d0b7e3e52ec284bce980645452c7f78da5f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "34d74ae1f5b79b48454f1f5f7696b14360f458b56420f4745849f5f66480803c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172080, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "20505b24e480aa9cd2d4b8bd76ab626f27ebf8185b824de87f20ba2fbb7d1b01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "b886cb2fbe332201b2c3939d2943e34f0abc7fe3a9cb60d46050ad296dcd9992", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "63dc3e62700b19bc2d5b238dcdf682b9f3e3585d837f98d13f94e7bf8c32ba32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "705869866d4567297ae512f3f405c13ee515180b44c0afc4c1c92f6fe80c7abc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "0fed17b59da4a1127ed3f22467d6c46d0ecd6aea8302f5853135686c70718a42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "749211c5ff48e9c201c810b6cb3bb66e8840032e5ad465b0706281289d5ad817", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172208, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s6_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "ccf93a8f66e04e1e0a6753a16b61465e1fa22cc1fe11c37a405b18b64112cf0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134576, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "3a6246b30614fb18d8a0a7a3d8228d1a9a88ee9d7749ceb3424e16f293ba7eea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134576, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "f803808b9ff7c3b07b38bb7b7a16eaa3fbe7b725fe28dce69a0d79e0a8c5a81b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134960, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "7c5d0bb94e01d612af8b7f8f889a4db37c55ece14ca32b6a1df6c6ae8d8af93f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134960, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128u2_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "57e226a128ca506e4cbb48c108fd5b3d8a0b870ebbb67d91891da3670d6f12b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "ce1caee6bfd31004b8ff8411cae8c7ca40b5eed95c1ebe606443dacc1bb7cd51", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "360b515ba507f8b70f7d8f084a80103fb1b5f43980cc65b36d394c98e8c818f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "75807613e297630722133f1394cbcfa2078d3354d4d129d751e3892ab24080ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "60c44a6a45091ec55dd758b9f62b655e760fdcd1ab101469c985f2cc19e0c363", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 110768, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "ed3d41807501f1316f21bd1b0de909c32461ead139c1c261beb31ce564f5c1d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "74cd1350b77a25ae57d20055e04fe6fe37507b2996ede35c7acab9d7392f4ae7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207024, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s5_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_noShfl_dsFp8_schPd4x2x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "a019ef43201c9345b81ef84e4c6f03925ff29cf8b808e51c2f81fd97bcd97988", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 0
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 110768, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "92c3f8716a74cd67190972234b678515a0fb094280fa7fc8f7452f0b732d508d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133296, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "8890c2ceccc0d36007f5081c0f7a3b7a560a5659025131473e07eb0d56116d1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 110896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "28d479e41fc5ee501b85f59a0db0a9707157d91ed5d365edb079a46c7ea2e939", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 110896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128u2_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "cae8ba7d3e36d045b0c9f748b7f101c5f3a452f335086d11099329d5fef68895", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133296, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "c7716785e447dc8f36eab8f83e58acf004ae4bd8ea42fb4ed2be9c19f6d6320a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "7fc75707bf094eafbe8229ccbc2dda246882570fbd59062dc92e57069cb3de4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128u2_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "cea007ab5929dbdd3303fa42847f5e66cba7c052d251549e7a55d45d1c64a825", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "8d63efbd8eb922278abe107fd318bfa57baeeda7f00bece6c677cb63adf8b52d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "a1798198b3848ce78481a1a9b3eb7570a4b043ca771a69957ec9b6bbeae2b9a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180144, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "cc1ec758cac118e563d58bccccbecc3a3f491f8ddf5dce02c3442e1391019cef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180144, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "cd18de12d7bc315952968529853b4a0296c11aab8793670c921e5bac32233d1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134576, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "c6540f221495de9ddd19076859fd04e508ed62e130781bd20f79ed3c26c0fc9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134576, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "266f5f5fab19e9511f694b1d672bb8f2896b6ad4706948d029151c9733cccfc8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 110768, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "8bd365b492600e437d9fa303032179d2c00c755b3c741009e1f0d522b735a202", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134960, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "b7ad381819bb63dbac2ae3aa551ba2a39ec7089650d9b2f7b13a58bf4c06fbc3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 110768, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "e107b8e3e0539ea7d66cd8f296ee0d39c617890c36203b0b26c7423b6f019fc0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 134960, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x128_s7_et64x8_m64x8x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "8702d74b98a418c80bfdc1016c4edc78cadb59410b707c18557b7eb2a1c463c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s3_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "3039e2002ce279b66c424244b6a9efd49d61d26f008bf5763ae3d2dd13369df6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s3_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "ba1687495b83cd5180d648fde4c5582c949644cb5fe03b7300ff28e3e30da8eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 110896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "5d88e0c860caf52bc70463637aa0396aee450689cc389d4df25d329de870f40d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 110896, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x128_s5_et64x16_m64x16x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "028643f032d3a1a124307759c4f2c8b8c6de7f098935e3b8f2138796e527cb53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133296, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "bb5b2e2429db948d6d5ed24a3094c8d5b01db8e6db4aa4aeefb744aaaa7c5e25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133296, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "972467c624eee67d0089898c4e781bfe4440c838619937a3f78df18f80e501e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "247af0c8dbe9d43593b419282ef5b5826aee34522e38947268c0956e0a287551", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133424, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x128_s5_et64x32_m64x32x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "a05d2b6a81f71814648071b0819f0bc771aaf33bd6607623cfb4e94cef1491f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "f594423a8720bd499529214a49ff641145280c7acba28e3df1c6560ba123c058", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "12c537cbdac5e73b53de25e833eeef84bfac7a2b437459c5c0a812732f861351", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s3_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "ce5f30fdc9264db6a61598eb697c46f931c98df5a85e718f094d7ea3700f7260", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180016, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "70dfd318eccad677f4c39c50198ba459fb0248e73cbac558d8e93cc6ab587e3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "dfa323337d176d15977bdbba4736b5ce57b17b09c0b7bfc6a01f6a86968d6f9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s3_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "ff42ebcca1e2dc3d70795a3cc072f236505a1fe56e49ba0456ab69e80aaed9b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180144, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "253229876f8046a9dfcf7fcac3a59d1a47a3af8830722d5f099024d9ae7a4125", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180144, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s5_et64x64_m64x64x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "f480ed46abec73d34645f8aa35f6c654afc7ae932223c17085fc93dd49fa45e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "8628ddd7ff353213aebd332c65ad92aec1f72c79b7a47f79e2217cbaaa0add9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s3_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "806674709c685ebfc0ef23d19e7bbd3420bb2d550198d06367aaec03dc4e50e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205600, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s3_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schedS_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 384, "ca39e4a348bb8f6dffc14176bafbd9f4063f419b1cfaec47dd58834cbdc67d62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "2d6f11b05046308a8d8a16bbbf82c0fbf024a0b9fa1a44fa44d5bc740934321b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "674dbd3c98a462d01969edfd7a91a56ae7658dc40e38675888ba650cbfc6ad1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s3_et64x128_m64x128x32_c1x1x1_rM_TN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "52dc4955bc969dc085060f474c8f9de7efca08537152f290148f363bbebdba7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s3_et64x128_m64x128x32_c1x1x1_rM_BN_transOut_dsFp8_schPd4x2x2x3_biasBfloat16Mn_bN_rgTma_clmp_dynB_sm107a", 512, "7fbaaa63364c537a0e568c0747e6fa1f8e6b69e5e0004371465e13051a6a2c69", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 64
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 1
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 64
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 64
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 4
, /* mNumStagesMmaWithinWorkTile */ 2
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 1
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "ff5c40ebfb41a681b1612bcaf607386d261f7c3fa4bfc2a49f674c22d09404f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "8a14021a896d240f8715b0c62ff62c02897fe11d05bd4f77e029108c50b7a97d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "d84baea7e57dcbfa423c45818d6c1520c7ea5794f617fd0dd648a612cdd6a185", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "90654a989b0e3eb3a9b7a8b8dcd06bed1a51db11cb73d9b19d0d3bda23235f09", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "5be6cea3b74045e0d8c96fa1d61859a8ed0796d811b064d43446683bc7ddcd3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "baa66794971f9939c0fe3dabefa032ab879f9c80e9c815e33602cfacb224aa2c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "e0b329cb68a289ac319e13522923924f007f6a939431fb08d954f072bd84f4f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "a6acfa3af9b3dd3f3acdb5fc58cf02b3c9e5514d0c4509db7a28fd4c4f4be67b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "cc4813c427f48bc9b0b12e03ea81e181a30125c118969766a7ad00e1d3aa8e18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "4e5aad9df8375d0a1923adda3f3a7fc62d753efd845d7ae3ab79e93c6e43e5ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "351501e8f4d16488876450055072d9be7d8852cce0cf03b2aef752e54be2fd04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "48fa7c55da9f501aee38b3c2fecfefad6698c2ca7bbca104724d8ff2b5da2545", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "1380b04dbc3b114709f6c23fd6fc1b513a266f9183ab14a71c6c44fa04ae7d82", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "32339e0dc55fdd3e6ec0250130014f9240970074ef0861d88c4818b73348eabf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "8e1898e0c71f47cdbe4b0e8010d21a2c09bdcb699a73a5ba1b031f3aa64bb35e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "3562b265cd3b97fbebc8fdb7257015ec06b45d0e7d4d94ad80a39f36553c912d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "15cdb373e5efc7beb2509c62b25ce79e23231b32c6fd6b9cedc2e6201596791c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "894c2e9182aadbc71e83dbd7ed053b19e6cf4f60a3af4715a2b35e7003e91f62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 256, "6e53c249cff6019d00ed4a9b6532c25d1b50f8d7845869f6fdc202f334c8438f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "78eb845fa43904c7a26c3f51c0e1f5be09ae217bc011fb8b4e7629ad21d46f0f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "c3daadc8afa5c9e026c7f8c627caea2cd010a1861c4d928090b97eb2a2f6b311", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "b123db7d5fde574e2d1fefaa538a33029889007f63039fc0f730f20fdbc86c0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "f78bca184e1f8d9e71fad5894c4b7db58e211b8b0a566aa52e3279014bb9baf7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "8c3534dc7b87a79921dd61f86f9bc347ea2aad7a8357393dc097b9a7252beed1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 256, "e38f4e49a8f13f1f27dce7ddc8369ec3eebce3cb71bb7b26cfc68ccd89a3ec9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "1b7e5514309369e5e6e96358f9135e6c4dbf56d7272c2b00d65663634be084d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "cdaccf5c73891d4cb8d7d8aacbfbf93c1e704e48935856c33b7fae6fdad0d440", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "b36238470dfde41c2513e39103c1b70a4a7b4d491df9fe7602253f82c8558bb3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "db67ff93430c40ad2c9bdfe47d011d10c59095b5e4f8f43f983f42dd288e13c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "69aecdc496d94a6081c7af495d96671b80a369626ce2d7184cd905e9a89f0b26", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 256, "174d342d7a72200304d6e8f6be809903af9990c6c6d145faa214e1fe2a973c42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "1ee5125c0639b88bdad54a504c3fa4f22b4a281f99ba20881f9b680c79cdc0b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "dfe95eebddd9475aa549be7393f5c6c956939e261da41d0c8c6667704a83d536", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "b1443e1073c17442c594fcd80b3cbad3d7dc3e994eb68da55a3702b1b002e91b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "5db54b89542fe057307ffc04616b288a7340701aa74c21104699a825be6161f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "50def88eb31444f4ae8aa5037c265625f729dcd5bdef1ea896f66a8cbf422b6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 256, "64ae924a63f9689eee868cf295794f499253a0ef0bdd7485cb0a6e498b5d0fca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "4497fae36069f09eb887e1ae244325c89033008be5078ee647307a9b4dd89cdf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "7400c1ffb04d32fe39152fe3bd4e93563e2caa13e4be41c523a84b3eb22c8170", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "351b972dc39f9b0e40b60c4cc7d7e845ef0084212235d7ef11aba6cca93447b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213264, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "e7889d226d5dd309cc1792a0de7d4c2ea5f07547aa094e0409ddd84bbadb5744", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "738353a6641622f80efd791846eb11123c22c7ef6c8d62eddc2f03b441d874eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213648, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "f8d6b78579068b0eea03a88360a5ff985f3634b4cf9b42ce07ac40e4cbade020", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "04f63f88405212013bbd6d635306fd77dd65a16958ad3414149686b513bb477e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "0ecd08f3a12553d3d0b9e42f881e27585cc289a652fbe9cd376a44ce56fa186f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "2783a5cb342ecd04e646a55585ddc42d120a65c80633f500a2884342e60d3587", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 640, "688895515ba3b7bd9080cef5c63f43feddca478060f97112bb80a155122acee9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "00c6df7853b70a5a50dff029011422c9e981e6b9e7df8561bf289051a72255cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "e69c8cc1186aeacd40c2114d9e54790a16ec045bf16197796070b84f4f5c0601", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "6b885f0d396b85ef42387da2eb2c59d3ae307626a076f12b48691cc9dd0adedc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "37df4c53a02e9b761eb33e9905a80169bd5884fa95b01703e2b0e275f99c3f28", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "ab8762b01cb2df70c8e4e4bff0347005ccaef29ca55312120fa8fa1c6b7c96af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 640, "d23a820163a2bd482a3c9e114ac83c22ad4541c4647e1a997014be6db39b6292", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "397df000f29169d239422fe5be1664ddaf4e27b412024426143b2b7847670a9d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 640, "80ff8bb59266ee942c5a288c935bbd2b15d92b823f8650294f785b60aa589fdb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "70ef7698e2b358c9835c5da6bf48eddf0bb9ab265bd0cb23e038b354b9828f5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "2cc44b94abc3a7d8fbd311970fdfab96980a0ac0b8a8951022afea6c92623f7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "d2889a493dc5d26b727c27e81a1d77546bfe7b0140269a3e019048f4e109ff27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 640, "57212516c1195c7e27df5ef909e192fbebd2ca11a0d86c65ef47e29cdf48c097", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "4cb6babbd364d04df12f57a19a51bd92b0d7e943ee49440f1a12b1559da325eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "3a2ac7568a0b6692c2e0c76638df3fc4a11d931c2cd4e2765c9d69ca5112b31b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "451459cfa9e3f986e198cf0b44397ad7f77c2fa9c7f76ca7ff244a1d1c24c1ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 640, "c62ab57a4e0d4e791b832b2ac0c0ea57a0a6bf2ce6b4dcaf06c84656635f066a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "688e1cd0a1f6eefb5a899f9a58b748d46edabcd78b20c80d15ed47f0e0de5799", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "b0a95648174b39d9e2197c38d68075df4fc0f6cf4768039e93296995cc101a0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "b7656d6e07f5192ec27e253c78d3bf5da3509960a46e8060081e74b165f289bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "1dc0ac083b6e632190ae6b8c61139c5ceb4577dc3447cc579ee59e0aa58dddf9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 640, "cc8b2cd39c4fdac8990fe10964f6ebf5fa579d9c14919d221851bcf15139fae9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "2d5e5946088de4157691467bdeb024c806aaf7751de5f040b8d9bd47dd989f9d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "a8d79e53a3f38f8c64c9842c02ffe4c531f56846bc0c154250a4f244010d47d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 640, "a1dc0c6041e3202bdc5521ba2ab36cd3aba787f261ed9b3d2b228be7230c1008", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "34274ef3d1e2d47eacec28a7338cd1db98024d895397dccfb05ae702f88d71d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "5e489dbb1b816f791bc073cf6df3345b818549c59964089a1fcc54e334c4da4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "cbfae2380ea565a3c101f071eb9ba9d919a5c2532055d333fe4a8f8c4507e47a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221848, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 640, "ac26d795e48c9a8d3e4aa8a808a0c2638f43e73261a1ddd7e54ff7d4a3f39191", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 212632, "bmm_E4m3_E4m3E4m3_Fp32_BtokBfloat16_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "f73fbbe54ec54a041214678b9781ff269b218e6f425782cce87b8a843e71b586", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "0e873322370e93f9fa01adee208122d85d2e43a6e2fa28bdfd9be3a6735f18de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "593137a9648eda859d3f68b9842c9600eed69e54a6b4b59581f0b59acae79951", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "7fc200f13f6344a673d73347338f9f2eeef1668afb91b57e765d7d0eb5e80584", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "0d6b6df16dc4822652ce51245841987560f2a23cd928c90d27327a2a43022460", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "287961ac4d1007171b1aef3809d415e33c8ecc28473a3fa608baa05dcef78aa2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "aee52c0084fb220c5b02e846997b2be30e78964a61f130bdc36081e2ef213dc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "dd9cd5bfad8ac9dc0e57120f720137d3cb6f7941c06e08e436354203c4bffcc0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "4a327d649c4c6b38b81bb65129dc7e0b6a545c97d4a4e5ff512ffdd6cfb88aa8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "412591dbe5fda45d7ef5ac908f812a168727ff676a3ea2ecfeda0c46b15e799a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "268e9d6892dfdfef1649c63cfa6f9e2a759173ca88b94a6ae419df7f4005bbc6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "39747f45b340adbb3f02167d9beb62da81dda981666da470049b6dac997e7952", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "dd34b34025e52ed1d812d28a459c41df1fe6212664776cb48da12a7b46add651", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "04590cbd51a228d1cf3553dc4b1785f64a6ac2db8b076e6754493170faada1d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "7e3dbe23e933b1f156484d1f7b8d1c71824ba19bf88db6ab091f946ec3b167da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "65118259bd8ed1faf4d00c1c7f0fa1864b28ad4bc2596b039773a91c21362d0b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "d4bcb81fc659e9bc28f88b184dfde0492ad031b945be04148ad2bef0fc1927e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "50bf37ccb75ddc73a832c20cbf9e45caf1d25fb24eb900bdf63449179468e076", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "be4ec2d8ea5d8fa600b27c68fe68d3b22cbe2a4ea34e3c5ede5b4501932c8d47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "c21e70bebb0aabbd627f8bff1ef20e9adacf294cf4ebfb820a398049debbdfcd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "57bc2febf1e1b41b367b66be85f5a7e9ca938948767d859b8e3cf1e3d1df75f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "4f3260bb97bf1d7a5024b98cbcdce59661cb56ac4da0cc4148ab4d8414f4c7fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "9b5232bb1e35f40d0b5a74b6739ad9968643dcc79eb9397e5ef0a4f6184cf3b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "03226a33f50b3746680ba76e8825a08f332d14288f74d66f461c7b56e42e5edd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "38d06d6d51b7df1a4dddc01a4df3e9e535fff8d3c54308f626e55e316708cf48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "c92335a968daf76e42c6008a7555593d710cfc3fb4e512f846c115219ed2f645", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "7e6207467345751ea0568a2a4120497c40d0cf83efefe4ebdb86b15dc123774a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "c5c076e28b78362207573f3c67cf55981aca2f47e24ef2ff4b022641963822c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220824, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x256x128u2_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm107a", 384, "52489d4d1ff2f13c1cea00e21aca2c21ddd17d2fcaff7c2ed28e5f236b156d60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "03f2888ee6eda429321f3bddb78a09aad29d3ad1296597b0480c3bfce72f3e29", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "66eb36a045ff4b3988a81141e69fff971fb3f1a4bbe32aeebeb3c1973f4790ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220824, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x256x128_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm107a", 384, "c5bbe4d9f40c6f29405e7b16ec14fb88d7c1b29d9937bc93fa75d9dfd7ca7d8a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128u2_s8_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "ef0a4b772a1b09b4a1827b3a7e00e826dcd1ee43d31a41cb99c202c0030769a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "14dba5e188d4b4a5c105ec2b6514ed26fda3f575023ff7ce1900d13af9c8397b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184984, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x64x128_s8_et128x64_m256x64x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "987c1e4effda15f5e319f3bcf76ffffb484f6c1ae41efba1011a1e37a4c6177f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220824, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x256x128u2_s6_et128x32_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm107a", 384, "33afa29a4b0f5ca8319ba4debb013c0ced687739a0279258129d6e2d4795f4f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "7548864f2e790a19844b9e535980f251d9b08cb92b1401e71d36bb7f23b1a1c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "1fa1706beee299bd05397ab5b8e593c7abf27d23f906bcc614dc6839c8a32c95", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "30a6169d290e0c1e67715258733e66282b771230ec999d4a0b4ecedeb9055e11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220824, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x256x128_s6_et128x32_m256x256x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm107a", 384, "a515903cbfc35afcf0fd63ae10fabde7db56c04687898e0ead63a2439f8e6039", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "693166cf4e8fb1d12dc5ff7a735347fc4b3df9bf10dc551db13159022e63d7c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_Bfloat16_E4m3E4m3_Fp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "a43c9f20079a2c3b7d231966d27376b623c4be3d3dfbba0aefe132ac9d03d5e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 128
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "900b66fd8b8498a00def5fef8c111e72a199c422e11adcf73c4a4417e95a9783", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "bb4f87b6039eb30be45215f5ff05ad8756f48c3b12e9f17528a3bb6493c08ad1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "43a2b7cc6ff43dce1b1fc652de6eb6bcee1c31c7a3e29bde4f4b48e98914b811", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "f3d742a8f719ced566691e46bb2cbf6059ef2999554603fbcdd7887c5368596b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "5d9953b015881e1c8b508189bf95246e98ee28b20c8e3be968b1b6ae5a37914c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "39d7df81d6aadd0d5b8750b957f1ed73ec37ad34dfafed7d49ea3b949103466c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "77dc3bac5f3d142b68150de81365e989f7ee1592baebd8b135aeb6200244bda7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "8d225d7bdb0be7bbd4df100e0f8d4f0794a4a7a36d8826d3b66c345f8bb9d178", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "0913c521b0cfacc61c79a5398b6529457c85214e6e22b19c7724add8f5b0bc9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218384, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "a8e55bcd20dee2b48a0511733e4188ea5b801ff712a33a0f2acc37cbfba8a843", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218768, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "34e0a3d30a1ba447f2fe76f58a8425ded1d79cb7e931f1ea09119df110192c73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "192743742c92b0f6e0a99207c72ee74c11ac0aea64d500f6e804ff99b25e9ac8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "80393768867d2c27c26a4a108f3c3415eefdc2fa0caf093870a2cbcb176e600a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "76882d803bcf62eaab5c5ef231da9584c24fc2257af7f893cfd0efa67d819647", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "61a0ad9485bc3519bab25b7409d465e1f54ae4ebd4c6744cde6496a239d005eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "c32c38ac8297b9defa2cfd470d5caa67680a1a36daff321dd3eb132d986dd115", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "5b76d40ce37ef92617e20df3da357a029d35ef3539fa03cd469667b1ffc0ecb5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "472591a9fe15954628913bd0d902a42caaaaffbee048c4ed31b32ee0e8b7303c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "15d89323ad7cf7149f7e78f64fd17dcfb8b80bb76999135a75a45461ed920a96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "a59aba740b544d75eb22f4a20ee36b02f739221e897c492b5ef1b563ceb50bf2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218384, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "42f67d249c053d4d05a2fbca12b0b7c90f6e5e24b236ee13f6166bfb580aea5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 256, "9f11ad80240e8277eb0b444331b1706cd3f9b1197c6b0309982b0778277ebaad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218768, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "790480eac017f51139a21b2eba46b2e80e6e917efacae12fb185eb946488706c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "3a967496efe70e3bd8ad0b6e9feda2500f4be323b46b6c68e05b321f58c0112a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "c4c17ca873457cb596a050a72aead0fe7a7e9ce7af3e36c47c6af9378ca759bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "25c6b58af624fd3b276c17de94322a99b37631abf86625f9c6d6fc29dae5cfe2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "d1ac90f233e18930e74fce611cb3b7dddc134f6428ca3ab8e0065cf2f72c1657", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 256, "7d923e4b6e2ec19eaeffe017e2c11d2d06c331ee0f7b38fc3e6a4f095c537a63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "0ce9bdd8b4842e376ca94b005ad6b3b64d24e5469ea91b1815a04aaa3fd7298b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "1e1d01a95cd4e817f0ea9814654d91cfab3ac3e923ecad6b51b376c66aceea6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "b4a4b640e250011c81f5c4f8dad812c89e5e910f3004ac0d8b97f622f8b30130", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218384, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "7c120f8c43329796054631f3e164504514cc19ff6dab28f75ffb4e9def961422", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "5b816696802e35174a7489dca3f30f1ea9dde85c2001c14f03f06e6be991da76", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 256, "ffa72338cff52cf5144bfa3e5680284a35abe86b36a0c73b1e868a3379e99ee8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "2d8c19bc8c0a4151c095cf828886fcb869c69313985600a95b49e5d95fff7dde", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218768, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "d02932b67a9fbd5c1db57292a4f9cefa717258308e0749dab9ed550c92058e96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227600, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "c76b9958ec7ac22da494bb14ce1177fb8dde17c02878162570ca72c5e4823826", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227984, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "26814cbab5a6f7b67161d72b6050510643acb84bdd265fa24ab1e1dec28b1d42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214288, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "06457aa5e11ca390c2c7cfcd0dcdef3cee89b7ac153c754f0c583fbcbb69d4af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "7ca03c23f9ba9fd8a3617b0c2cd21549c2368a7e7741cadb730169a93330a07d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 256, "bc3876199de9eb572591ca5c9efe099af554c16cc62b4c8ba297258addf60dc3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "147340e10ae90043c8ff16dac10a14a5eb86b1e48e31f6caa281ceeb82822ed2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "9aec02e0f68cb12e72def1f06bf43e98e790fe6f1f1f6f857c76495059a8fe24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "4790ff28dbed33f1ecea6ef5f7724e5ed10060ba687ddf5be7dd9a7479ec05b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218384, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "f31a1a30737b051b20fce891f8b836b33f2d62e00c7f72d0de554b99ade4afde", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218768, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "3aae2bde01536f7038b7d13a6de9e0f9dab904be7fc5c7061405cda0e0596540", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "4618ba46a92ba080f40bc331c6901fa2db628e9472accdc6313afbec0dd82488", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226968, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "eb7962b9d3ac23fb373b22c5ea3b90de352be9cd320344dabafbf33787ef5ced", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "5af78a42a00f68062de3c159c00a1adcf85a4588283857710067636318306e1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 640, "89e14427d3f63bc44dcf8fb8deb5bda20b6e8d25352e8dac701e87714c74ca81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219800, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "3b0597bf14ea6740c4b69b7d305509779eef1cafd0bbae40c1e7f46ce77b26d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213656, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "f1e4b0f81998f407e7cff1c9080bf6bf3efd8b250ef59d45427810e572996efd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "e5375756d74f670c1a6d25cae97e00fe38f8dc45b9467ce2ec54fc6326303368", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206488, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128u2_s5_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 640, "6c99eec6c3158dac06a7e705f01a6126fb1ed4506d618761d3bec18009ab3eae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226968, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "7be686b5d748594022035b0babd1b7ce408c635c39701e10fc4f45b8c98797af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "207d18953fff7d812ac59bf2fd6d62e6084d75051652911dc6b64edf106f2445", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217752, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "3edffeb0396f443e72a62c8089a859c2edad2bcc4163360118f01dffa6281d99", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "ee10f0c5815bd90250ebfdfd18893850b0fdb38773b0daccbbad3267dfe9499c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 640, "6b1a3284c30902d6b32f8da78f8446525d2507a6e3a5501b2551d3012fb7739d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219800, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "4752fd32bc52bc97b1f03554d04de0674a2136c11a712796494eb1e48ccf5d0a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "d5dfe55e36103acf0ccb2f5b4869f770fbc89a3c865850a0b9f86b57c3bb10df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213656, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "401b3c18dfd1b3bd09c7888836f9f750254f63c4f21a449c0498a6afecf63950", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206488, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128_s5_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 640, "c576945e7b737fecda6a236e307eb4d1a2abac2e9ed52ca716a88f0f775230df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226968, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256u2_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "ade173a30cbea3c4e8342223b4733730a33a605704e22366a52735c019f5eb7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213656, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "71a276ccf7bf835d00d9cc7ed07f534622caa503249d438fb8a48ff5b7f42342", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128u2_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 640, "aba4c14a65a658ba75efcd67a427369e9c48185a0872961b7630eb6824122059", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217752, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_lbW8_dynB_sm107a", 512, "0be6066473f7e91031e9d1154dcdd848947b9fd0dc82587d8705c70d907761da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219800, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128u2_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "88d8c6d370f949b60a951f9f7067fe434564d00aadc4c1f74ee8b6fcc0352e3b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "5185c295e3914cd67b395e27452df084be04219c89d0ac780b1a67af7321cf2f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217752, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "71fea6fc824af6061b7bc68f58f6350766c511c8414df14205e2afdf73a1309c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206488, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128u2_s5_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 640, "a5027047047e404f5f93d4c32a5d3aec13cb910c637b735fe58bece95fe297b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 211608, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "2189f7b804a0575297bd2bef41d4a5dcbaeeb176e807ba751ca22552f0d6625f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226968, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x256_s5_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "35c2c8090806e44ef1a655701aceffd999b22924cf3bd96c522ffd7e816cb614", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 213656, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "926646e6e1f9ef005794c41222945cb4113b8a0c0c28185bef1cde535d88e84e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_E4m3_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128_s6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 640, "ceed7951fe56f65f1a5755a97a8584f68dc75c5e3383e8469b6c5a53b6aca701", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219800, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128_s8_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "8989cd7c080a0dc900e391fcfc2328fb73c70690ff9c03fcbedd62836e81685c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217752, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 512, "7af382c278aaae5167e49de2867c463f0d5b99f85b88f9158414b551b266d629", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "d97bc4e92bc47dfd8ed5111c33517ea457fcf6429e55b128f8d246689c40e5eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 206488, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128_s5_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_relu2_eW8_bN_tma_tmaSf_rgTma_clmp_lbW8_dynB_sm107a", 640, "aedf94962a3f1ef6e7fed2e9822f163014f78e42227243547ac2828d0811339b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 56
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "717e51f9a101983124902451f92ff2d3dde2ddae9d2f210d5c36f4196d2d1381", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "acb86cb9196abcaa2b97f2283706bef8341df6e348b2fc41d687d0765712522c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "3e55a6525cb9d4f887f1eb6acc5c42902f1567e3ffa818844b9009371bdb690b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "da65bb9eccb4cdea1d9dab6b9a758d4bc0cd622e5cfc7fd68bf6470cd342e63b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "07459664d4d544050faa69b50ffeef9c01ec5a0b90f4b35dff9fbadf81175588", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "a648475143515acf85dfb4b27c5864e5d8cf9a92be5d44903d6f5edd5bc2493f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "4239febdfc611712410a7e3d57d75c3f470610ad5ea48dc32e02d4c4ac1ed405", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "7a9dfecc9d97efb9dc24c80b71742cf9d26cf562446d7a4cc2bc15fb14656869", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "9755eb0510bb829b1ba74ef5be7830e5d6bf5a31c2d5a4abd7a55a028f75e315", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "4d58ef41ccafa02438856fcd23c155d804c9efd664f4a78814ac14de1db515d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "a55d4661fb824bba04fa21a2bb80851ad924d6a9916a7143d43d82d2d1822ce5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "b0b0c3f27ad801c75b2231d2c1e88dc733dd616a214e24ca88fb62aa83235c5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "b162949414546e0b48822a900543827cc1f47612a4858dc23f156cb1360731df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "7e94777434d878bdf91f0bd21e69a50d3aa4b073ec8da83bd243418dfe4d95aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256u2_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "57a0dae6df95d61f93c72aab838c22cdf071c631818065f4ebf66eaf09a56166", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "f3fbbefbb993cf72ad7dccef06af77707a2d614024686822d454a197f6633ade", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "b316b2a63960c7092b4446cf53a52bfcc1d2ddfe6376ea56e94a22f600178667", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218384, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "660acff25e3360160e8bb6c265b2c30c57e15a938ded01c717392402da445267", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "1f05e707419dbb85370237d618b65af5ebbb56517508cfbc0fd3d6ad0f6bd799", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218768, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "e17e920c3170771ca3cded8c269cd413377aae0beaad70eb856af84e5504efd3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x16x256_s6_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "e06f8cb72b2c022240fe448030ab70979ecda55475a64c183605b1c95ca27767", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218384, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "0478f04f59a9e7b92d82003082de7818704a4822fba61319f48defde13bccf6d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218768, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "58dc7cfe9d6e525058be9370c7d8fb96f51e5b1d999696fcd1065fc7984cf8de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x64x128u2_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "a1520e69ee778fd52a005d5d7b137dbe81aaf995d5e5b84f1d65c3278c8198b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "75d5663992f7c80f5107a4258066e0675cca956bf7785d2ff2c720ef8106553f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x64x128_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "8d50c1d1f2edecd5eb5a514251aaf246329653959360e2bd50baa0bdaf891b5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221848, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x256x128u2_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm107a", 384, "c02de7b65e557a40442b918d94ba26601fac1c3406ab9b969e11bd40318a4785", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216728, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "2a83e2c32e64933097f31070e3d094fb2c19ad97b72d1c050debe0bcb2962672", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "0c62dcff3bcc032322ffc6f9bfe4d4a6366a288345c0b891990dee260c4dbf10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221848, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x256x128_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm107a", 384, "a5b37cce05f05c05669952f1c7e6ed82168c9cfdebba8b1ce4e2d2ba76a7ab02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x128u2_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "6fa538bca67052e75a61244a7731ce7cc105f03177091c9c57aeee371d193295", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216728, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "cbbc7dc690390865d81660e3664a90a86576f4c6b675fa1d270613d4d03c6339", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "4482b5a8fe03214c2341d410979fccccb22d47f80aae3545f71205ae394cb65d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x64x128_s8_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "baad8326e79c0f6ed6902f8262728f7858c54f34a9c2d2ba65379422100395ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128u2_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm107a", 384, "c65ab97a43da0e94e3d58133ce174dcaa1a0e689697bb4ac9c85f67f2f9a4f96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217752, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128u2_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "4e07d10cc8a688ace74bef354a6367547cdd2c2a93016baa50c4e313617a02a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221456, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "c7a69a0898789d089edd570ee5edfbb1b9e2a33ae1f9793a0ec33f8cac48a000", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187032, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "4f4c913e98d518d16d9106d96a0b974fbdf8eab0202447d29ca02dd73a73844d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x256x128_s6_et128x32_m256x256x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_eW8_bN_rgTma_clmp_dynB_sm107a", 384, "60fc31f5c2c2880585cf0a0e233c837c53e522fa04f77f68df8c9f78d78afefe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 8
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "782b401cd937e67a931e34607dad4646ec6cbaae66f1e6fdfc8626ab489b3680", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217752, "bmm_Bfloat16_E4m3E4m3_Fp32_AtokFp32_BtokFp32_t128x192x128_s7_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "26acff8450914310823298f662e7774a3cd79e0df2aca355adb4a34655e7d1d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1050629)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1050629)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(2)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 128
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1056776)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 1
, /* mUsePerTokenSfB */ 1
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "5befd53d825ff0c576a7e883bf18b5e6df852e3e39eadfb721f64ba3d224d20a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "ddbf2bb3077d78d9fd5b1b17cbac49e630b2df777d3d44560a272cf9a762f839", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186000, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "e053951ecdd01daf456b905232ed46339c51f38005f76af8c948becfcd3c9440", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219664, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "b0bcee19d949d7c961db64a27a78be6963851c444ff7766c2920105c70f09097", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "4932f62ee7698fd405d70330df2aed1311f09bc6ded0b1714c7e7cb04f70689c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221456, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "0bdeda5871dd31b9328dbb0903c3b30a60790a958deece0167b9c417e427cb78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "9387d7d6d434663723bd1a025a347a331840991c4024125bda1350788ba4b976", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219792, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "8aac4fc6242e4af8baa73ea416b49acbc9819a6e7b24bcfc8626174ae4880eb1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "6ff767e42eb7c4e048ee2006df48970bb621e9a8b3699690785d6d597604acc1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214544, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "f9b5e6faf0875395bb859eea1e9f842cb1507564d106db833788f7109beccfa5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186000, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "86b3608a162b1bc97f127de85d5bb83ed869c30cc7629547a28693db0676e6a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "2be8993c4383c7b0ed5028d454424898bc80140a42d307f2740f64dbc8d60aee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "3154736115e130cab219cfc8a7d95c55ba49028bf285ccc2a56709c7100c0f12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "2fc9284a21cbf5653ee0fb5e5f9880526c778e6529a9eb7998ee08a1151d41be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219664, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "b8cf4c76ee42418c66ac906fd3737d214c14351a3bb3ef756aeb8aaa0e1d43af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219792, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "339131ae0c266d77320fbd38dcf0e6216b3b23022430fd67f2cb771b442063f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214544, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "def3d10ae2aef89a97e2b9322902f71ae1dda8d6e6eb17377324b00f35c8e98f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221456, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "6961ea783715fc1231efc773b65847e2262715ce15fa071a7f28c54ff6f3af04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "6e88a5f6382be11d33f182503c19947fea4c6d7532ee7a3c97fe822bb19dcd96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "bc30e3e31f8b08826535c6cc038db9270b9561bf48fcb88dba392743b2da72b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "eafa65e6d73179b9141d98c8acfe1d3981e0409ed13ec3797b13d57b7801b660", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "3e174d357d9fe9540391cb0cbd60db6cfd4babf76a44689cab398342993d3640", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186000, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "35b19d3f0fbbb0d2445d3824d0b7d906f9d8f864fc05b723ee09c43c99c22146", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "649377b086c97bda667f0e3077cf527d7ef7e4a83852afc49523d547f6dddff4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219664, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "83c3b5985053f868fd3031b7c8cebd59d2704473f5786feccdecd8d061c40f72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221456, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "082d14fa95447db7f7d6718360a71a1e748560960ce0bc6f879ee29a20d14367", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219792, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "d73cc359f5fdb1adf361feb71ae36983ff16de77f691f79081991a306a65c71b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214544, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "1ef50865f805fb0e88d1c663e08c662849f4faa8c1597062a8c795cf4f7e7b74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "3d0dab9481bec808f2877d211145b9ef643f00cb75d4025964622c62419a2342", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "6737d4ea471e20990fb121954b163b075870edb8590acaa3b1662d6d6b48002a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "26862c2e0a3fe292746dcbe30ec6c84413218c6a22bdeb2d13641970c16c5649", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "3c609aa4ca768d64816808167390828230ee26900ed69d9557de73ea91d4b5b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186000, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "3118afc674c217fad7997a5f5bce1f93e8d578e198cd67ba291265be67deb9b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219664, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "a3b0a14f7284930c90ae3af86af76ddd842cf69d66185198fc26752d39e639da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "e6ea050dfa638cad0b9517029563110d24eed8fe82671655e19485e3e338e4b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 219792, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "397787156019cf84dfa0c8171c895299692176cfb251712419471b4da5ec8e3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214544, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "7320313fc0dff1e4ad1e4aa3801bed254fc28589d28aa52e626dd5109ee1243f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224528, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "2b17969727c5989e55e2fb1c20caab1f51c1cc64992a2f0784f4ccfdba5c48fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 214672, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "30b04511c8a7b0b879e73bfea6fee2297ffb8a81709d7781bfd2f9371fc56134", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188944, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "5563ce065b63e9a54d5f51d562f49f78a0cc325326f6fd4d968192dc8ad84ecd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204304, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "775451948e9853563290598cdb69fc7c5070d79348f2135da0d493c56b953747", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224912, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "fdd4af1bf68e8537f5ae3eced4f92dc17f634db349cb11d909e13ab291d04314", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189072, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "6546122fdee6046b574da6a61b1ee99bf1c3f52e1b6484cec2aaba99e805d90d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204432, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "953707e618cdde6432a862f42385f6dc3bc8b301a3e6b660fc1bf3460bc980f8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193040, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "96fe37dd0478ac4f4b3c6b2b5dd5bc32303e48cf1fdb2a475be30d6726ab1d93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193168, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "d4c4670ae14325f392728bf64c59ad7ba8f6f80a5071f27a748280b4fa8e10f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196880, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s3_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "29844b989b4740928800f5588952e64a03649ae2b7fe985b34b6fb4a7f7377b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224528, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "e8b968c44a751cd26e8750dd277509ce306e9ad51be5a2d969eb979e769ae8d1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224912, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "4e158a44196c8bad4af13f411d2b1cd9c994bc142d664717104ef05171367aa9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188944, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "026500e50254b18816d64048ba6a97a2c7e4dfe386ed966e3e31ab917aba2419", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189072, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "e6d5c41e5a719f7ec9b2a92dfb6ab985bb7fb2e2b8ca3535311195250a625fe2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s3_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "e66f3c62350ece3c9b818e7148eef23a94478ac1cd453a518ba0652ae6ca015c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204304, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "e7cf0d720dfc9b487f20cd4ce06a75d7833b156445826efa940367d900d369b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204432, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "4ed199658c1c63eea94686872d00eb03bf5540a0054b7d5c6148807a72823f9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193040, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "724fa86b76f8053fb39b3b35b522cc9c83c1b4adf30dbbc2205df4abbeefa309", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "f802b882d46b6b804e24b5438188a78d448e9827fe38abcbcb7a7ee8575babe8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222480, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "9e7d69a967ede9568c5562a1a033a69b68619d45ed47d0ae7d7149def230c839", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "669a87d9b2898416e39c015c84998b0bb335831896536d9790c2e3e236207333", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187024, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "b0ccbe5b374dc3c218b501c8e9b17b1571252d50b3ae8ec0fba218f50513fb3f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193168, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "adaa535f2c0abfd3b4ac9ee06db84c115ccbb6db77717293c7963f01b47bbfd7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "255dc05572c6cc0f49457a89fd8fa89ce788130a12631a7371f12f527abd9372", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223760, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "3b3d082e6d158fb99195312b94013b5dbe158f312ddae73a14044d75c894bf0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 196880, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s3_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "73be4d74f3ff011180eacd7c9ad2991f32b546448e20774089da03365bef31be", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "e74e1f8613bf3db6a5dcdbfeaf179668227aa8f029e6802d806f2c6e2c19e371", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199312, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "ca527dde5284e76a5f26ddfba076299039a7bc5c611f0d19608fc09c108698fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223888, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "70613f63daaf737a04317e06f19f10714717a77102c0b66270b885115a39ee07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222480, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "c34b771e41981618fc91d7b4956106aa6252c84d5b25be07d2fc7386d3bd633a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "a5eb8bdfab8530758b253681791ad2686ae2185ddaf63d5ebab2cbd7b272208a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "5847c0418a8c18fe9dcf3982d7718982deafdc344884cc4d9c23b66806caf356", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197264, "bmm_E4m3_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s3_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "823cb22a21b07aa7d00b730a327c9dbf395cfb78f97fb5065d33bb1ebecdd785", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1050629)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "a8c45d061f5d6093cb53afd3031d6f021c6c70141f26dc176aea14eac7d6b3d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187024, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "ddcc0dc58ab571f396a50ff7bdeafc1aa38ab5da1acec3edb3929fe16ae6b024", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "092d1f628f51e997cf8f2da90d6ee5441f35f6e15bcd4ca212abddef8159741c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223760, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "ba3884ef4c571e0b039e9918c8385f941d2b7ce92d457ac9c84a3e2e9092b791", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223888, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "f8f43cf8833b5deb8dc56789672cc7013bba207e2fc31b96c35117ddb8578592", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199312, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "b820cbd4ef0d6b47af3400e23a7df021ae3249c83acf1bdf3593b0ee6e50a9af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "e9a6ef1d1d14889adae8fd078408780122f75baacfe5e0f0097d98e04a3c7666", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1E4m3_castMxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m128x64x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "8e8c232b425eb7739f326e1ba5af35046ddd5e6253223553a52eb49c816bb13b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1050629)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4ca703eb0bceaa6da990ac943af46c07518b7b0ba9219fba1719d37cd6e2ebd6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "30dc175a3943193204c81e930660a3e96859936d1e31537f2a40bb4ac9d75cce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "01628edf0257b412cdeefc978eb10f65b6485a43361076cc2e8f405df53e8fcb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "fd1e1d45112cd9f15f99f3ad3c904e9a47fee48839c44461eb382e24459ad431", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2739683bfe880374c6f1fdb2fedd207d8a6e9e8367cc46f938298572c2121dab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4825a027c887ef5e885df4167ca9a4de17e0680fc1121d95e902406f6059fde8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "23bbc4f0c4c0440315b633485d7187a219cc9b96b0ff1fdb9c24e0449c8e3bc6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ccbb485109ff88aeb5c5f628089d975ce836a1d605559f1e92c9014263d144ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6f926745d7272de038526f4cb3299f0425a1d8055f8bd92f1d002182f9eae866", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f9e14c0cff175cddf0ee1b84df24f06b27dbccd256c7cfb4917b43ae0bc9d2fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "dae2ba6a65bdd6aa9b85bf0878a3bc459dd18937e0424257ab92e2a1b2a8d4b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "d5a1fc2e582587d65555dadeb9b53c5b1346fa345798fc6ccfac516333814532", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "8690914d5b4a15e8b92e4c41a700f3a52a85111d1fa7417517f611a12cbe58a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3dd25f6634ffccfc566af936f5dca837fe80f70688b4948dff32fc3874d6b6cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "e08bfa123b83a3c4e6d30e4722345c599a1beaa3f02756aa16f199c2eee1e6c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0dfd75c7fb3ec23497dd4d922e2082b5c40ef6ae651247cb8a30c57dc5944e12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "d61d705f0775cabfd232a9abe81c94e270a2532f133fc559b0ec6837def0ea3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "085a8bd9a4e3ff2408e97ef0324fc3dd5c4ec00da9852785db5151a7488053e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c5eaf04f3834d319518bb645749f74bb07e7b5b99ffefee5ba68126dad0c5f91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "60e4b22a51656c0ba110e7638e76fc69c3c1e848e8ec41ffb667e1ca8632c96b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1c584567964eddd31462a0c7a237cd704db737a06a7b2e3405b16022633040e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "516a16e8e6764c8280a8c2547067b09e8338e4248eded1e24d46e77f3e0c49f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bbf93a1908f4b1e06bcddab6f94d608cce220a9bc00c26faa497d6f84ae0997b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ffb6333c047cdf8e055247829d1d8ac9f125a084e3983b4fb6586ace6d3a62b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e195823c57b0f458b0d2bae050e436d91d7f5a42a49fb3e026679671fa42d54c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b6a7014fae34097bd5bedb9e8b02b9f5c8fe0bba87e1556effc4e3228bd8de54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "08439d7282f2ddc8d847631d8160a4cfdb3330e776cfe5c8599d76b62204fa64", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "60c4bcfe3160749ffa265ff21ca7cd2ed5dee0df07eb82f38898ba3f3e246a7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "dd56ad73b447b5a312323f736005c756b8373950171f7827e94f13bcf73d148f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "520ac426f8d4b9ae4c60df530444a1005f6f9e872a84efe416fbf8014f009d3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "8f6908c5e21186d8785d4a1931260a148aee1d92b47f2855a53e713e5d8f85cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0398c633e67fe62277b34f7727d77ac1177cf9fbd2c86f04e05cf7dd5bbc1a9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "01abb919f2b76d6a0790b2891098d041da08966f98baba4b4b21ddbc47c37ce9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "4744335314b379ae36f5b513e49bfeede09535e1e41106f922ede27ce68c3269", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0ad90ceb85bba87f821649dbd8a7299eabd58ede27f13e5effd88396c4bc5dc8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "636fe71f48d252d29cae33754de9e0ffa1f939ec78a3b4aca948262d8859df97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "3660b3ffe33b9a53b56f87f227f20e2698dc5ec7e715c88ecd7518940146e585", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "4086dba0126ccc2800b734ed1d4099dce71c114e414e0223442cc6e0ddcb16ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4ea66509e2fd42d21caf087d538f77859b358b64a0ab3de8800491c93f783e76", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ec9aeea614ee34f20f5330bdb6087bd6e06102800566ca61ef123d615e2c4780", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3bada2693425e412dca1bd9e49d9d62126671be1b8dc14cfedce5d16a2bab390", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "49d7f1ad03ed0c778751da95d6b0d9fec0f24164e8d7a216ea1aaf9b9d91370f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1855eb6990ef95045e5bea38d8ffccf8059f2755cbf4336d93524962ded0ddc8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "70fa55555767d38094248f49f2fd11e461201948e0446fed820f82b80875e1b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "55fcc1c1b631d4420734b8d79eb85f6be89d99d4756a185e172472db3375966c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a409977b04bf9c94b62982f6711c3d93a8646f6be4c3469367bdccaecd4a59b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3738173606faecbc929d989b675f085b000739ddf29edc00c4a20f1853d88886", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e10866168825374640abe2b3a59fcbf6f674ff1c0200ee672c14992ad6c05ad7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b0e09f5c258a58db2e84f746ca485b97ee9c947d661c9ab091f561626077aefa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a17cce6d79be72da86fe672d6fe27ff4316c0c79b141e11ff6ecb433edbb641a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2e678d140a8722005017c0938739aa4c3fd6908e3d05eb9cd6c8561d82a9da2e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f32ef282ed3f44da2da84467421a7d67f24dbd5d5ae83f9c04085a0f57aaa2ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "55c72fd607a9191fff38fd17dfcb9afda84d5dda79f33a9b95fc87fae59409af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2b13b2286a118fac28c492088b1dcd0fea59bc2025629bf14f44c70a7541a1ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "110e802237ba9efe413acda45b0b68f1103d1fa1abd906ed7b8e1b1ce15d79d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "cfb37579754e56350ae7fe1c0cd16480c94f7795ce75c2189ef480c8e86bdf49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "635e29f9f61d6669f851e9aa25be363c1b59aa1a3d8d5c6c95fbcadfc0044c89", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "6b8581ff5098fdd711f5f006b8f468871ea50d49099879fe8d8ea19662f2e4e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4dc502f6978eecfa2a0f9a52182d3e28181fc6e1c61d3a2bce5cbc484709986f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "600db514648dbbd9675bc6ca2e756b85ae950e04d1a73bd50d91ead0c04dc301", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "db3208986bbcd3c1d9d876d485f17efef8080c5d59ee84b13cc3fa90eb264cb0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "204f5df454cf2425ee2e5e04f5657b10c2d775196738cf087bdb43cb16d08c77", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "2a3d72aed2800f91589f10a457006cc829cc30efce478ad88fcf576b623addad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0437c38a8dcb02b0b5acb5b9c9b82c85c743387849ab3ebc3792dd8cf5579f88", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0147f7302e1c51824e4349d2223781b2699da350db7d53800d983b5f5d99e45b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0c8fedfb1989aa3ae17f7fb32e515f2ca74dc7723f57f3d4efabbb1b4f451081", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4f906a680fc8e0fcf35b07a1c5701d802e971cb01e424b8b8bc602e70ce0377d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "fd1d1a9b1b3b2dffc09db9ea78ace0c5b7424a4a9153bb93f168efe008176163", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "68225eb7dbf0b062aecbc543918f1579dfbbbfcd7e60bc2bea3b16678d47ea98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "345ee217b060e242cc3efd82fbefcab7ba56ec02c4f9f5127baa4a3cffce6ff1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b49958fbc8f154f54d1d3cbe44a689d2cf8acdaf6cad3d137088398e2d6bcda7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "cb9a776b5a5a2fde042da87bb46b9927ae40ca3d8a5d5b4a681c65179b5e9a25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "1132b70aa85a431466fb026eb3b3a3883031c5c77a3a3ca533bcf168f68d3590", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7f2a4f865efe415e5749613eeeac8d8507e1b970c2ae201a7567340145d2dcfa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "073a019fe11fa108c6ac60de17ab5eac78120c07499f364d728ad7aa6fc904c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "517fe6fd6c39d6b57b8a369771df0c55d0a4c6ea94dcc47c298b819882c9fe2f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "2ab4ed9408428f40e6317fa76c5f397e849af7fe48865d262e0fef98a62a180c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b33d3207e4a95e1b35aa54dd0d80838de54925d0a10219edaae0237e55a55c16", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "8d2d756ab53a7cd1b0cc511ce0147bcb78139a6375d10ef00fd9305c70507954", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "c11a68ba2bd57a150ec124bb287cf482aed085039d047ac039fbe1538b426f52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "fdbffd6dfd78e0b219ccc5a99c8438cac6d6982228dc3502b9a59e3dde23b4cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d825d3c8164a5950622ef3549951540ee4e85908708a6eef8e06bffbb5e5e40a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "32b308136a4abe14fdfa28fbc49d396763f8ed088970912ec393d4852574fcd8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0e9ae67ed5836f0e06b674e34a5f7d01d0b250f3d7157e65fd0445741fc3095b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "31ff49e38dbd6e154a9de5c63382390898a6279503a6aedacd0e65347756d7bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "72461c1d39be0289aad0a4f8f7c501655e4ec559ddba07fef492f0cc5f296f76", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "05c298bae221c8699dbb4daae5743838fa96c13c1d54889ad22a2cf81b88b1e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c14a47ea89b11e6f6de38f823baf2cfbdeb4ea0b30d909251130f2ff97ab7e50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a4f4d2106f0374b1cf69926dfd447069c10eeeb3626591ccb76e9ca67b55cd92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "616701de174b61db08fa5ac0f93c6b9f66576ca47b0166c907d07d2c61841f46", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "5b51b022e416b4ee38883023e6c3dc821e503223ffc4350cbcceafbed1ea80ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4351b03f0f62d41045a11e3325884f5db1f1b72664a19e2b45e420ae5717b67b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e74ce5facb8d952e9d2047b249548399824832b331a5ec0725c26d47833a394c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "f0c3b4fe482ddea8366af84a18592d5b705038d44358cd9f461772c3bb0a4856", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "e7a7c40cbfb5c69d93fed1bd50f4c352d88d51701395067d3e8674fb65b7c1c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "28a20fe26ad2d667621204c5af08ab73cf4a14ae81ca1a5a4cf05a41e777d4db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "c8ebe16a031b97fefee82b57cb7c57b014cf835b1c3d71719620cc447b45f987", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "f97accf7a82383a53b9682d81c58378225151265f5e4521dcc0dde1e82df4588", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "47a75500a1867f98ef75e9adae4f379c384fbb22591dd714061b1f7d0340fabf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "03cc3a19e2d6d4634a4a69a1f9918504da7b8db28c6d6b3e7553e51220dd8953", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "9202e372cf12d4683e4a30f4ae251b8c883d6a7f6c042ba4c9d08d5565274414", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "3d10f8253426d1015ab20f78bcb441a8cfe0dd90ae751de0aa2d6502a5c44ff0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "741ad02c4fab6248c5e0b64fa4861c6a8f340229ade60e0bc19a90351083a33a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "c66b2c5406c2e5ec0eef1a6542ecf3d44a7e201c355339f53a53a021aebd05d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "6a7329e9cc2d23fab660741b7ba2e3fdbbb152ec42cd73632a0e3fa4b8bfa668", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "f2dcefcfa4739fb646806d104ea97913c82268777a782575434be445e127de94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "cc45f811cf8f0b9945e6a3df8087cfb6067e4038cd2ea40e2147704f7f30df41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "c333568a975b9b34f185bc3ac9f9339ad4d56aacb85b8f104bd7221f2635f3a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "cb0b1158e04e5001c70a93a39772c19df41fd37f24054a6e45327f2856f76df4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "c293905e41a43bdc9117751ddfd8188fa6496f2b25add7bdbfd325e4d73e20c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "def45d5ff31134a380314e4ac654bd0c66393f45e8d28871cc63cbfdf3ccb7e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "a049a47ca795230b7aa484139de2c29008657659d559a7a57e21270781089c0d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "a3af18778b4afbeed592d8d3dc7022bc6b38a4a87213564128dfa9d7e5a178bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "9363ca13e4a4584144644451cbf58a1d3727bbb5a5cd3f25de73c2c7441b0fcc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "87bbca2498b09657ed4b63574c6cde6b0c0ef453d144c9ea710a82aaafee8e20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "b3f9d820798d9f66405ec34ee87a3a721bf958f36f65e63b37b8c227f9f15a5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "97661c0dc1a72ed679d4d9de2983ff7f95160d0bb878881059e6c9590b465e55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "37ccf9a13f88e574b5c0c48b6e491f2b050d536a31a828c595cce2766c38e1fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "e721c75ecf78d3eb1e5eef8c0d74ffb77cb53736731611fdcdf7744ffd93e4e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "6b67c5c89673f10f5bc095d1eded75d07d098c4666e5bc9f39d522804b5add92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "7ed028374ab7ad09ecd91148666d86de1a50372baafcfd1dbd94067f2af651dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "e9ef093adbf92b5e11096c2991f504f006c40e17114dca075984bb056e9902f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "12b591839ed6e0a559a11ac6d6cb9a29a28fb7f03e37c902c7f825dc449d10e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "6f6825519b211ae6e1952e809b4077fbfe66e3570b226cc941c6ad84575f35e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "0e6a33f4dcc65289d8870cdc958d9a68b699b4815aac8f73c81f03107f24c8a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "a9f4ef9bcff75c5dac8c073f923c0868ca361f328a697bb3c017ddc8fc2baf7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "5d49dfce88297d3314f3fe35da1a49f3cc9fab7a2bc381023ddf2db10f4fe326", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "fdb1e33d98fed72f2f424dce288cefba5a8d4f819977b7444300f48a5487737a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "b1ea544879694740adc38870ed3347a84a015d1b300081d21688750cb7a69610", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "0bca6caf9101d0483e6988b298188d73b3d4cf1df1ee87a3436849ae99f51471", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "66280cd3cc78042d144da818c396d95802db41b9d3675981f556f77652b5a4f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "ee8d1798ab54d261dbc8a5e8d3643e9a218f78277f7cc172e0b8feadd8dc5290", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "1e648985ca607a11ae29dc5fcf856d693a775eb3c385225103db0640f2b172bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "394b105f67dc5dbcdef423d5a50c0413edfcbd726dd1128773ba6ade6aa3847d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "49a4759ede92fab1208592f0375d870f7945bcc22e39f0983909e8fabbae91b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "47249b466c5206ea4ae71129b80ee716cae35edbce7e5f8e291bca0a85ca40b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "642d34ebec3f4d42cbb82a659c0953b827d45a055a5e30aed90ff43e4afd79f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "0917aebcc7c6a2fbbce58eadd09af7e6b00eb1c915546e995a3e63b28e3c1a5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "e30ab7d5778c96596bc9fdc0e204db472fe63de25aa43232aa48800437529ed3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "02bed671046e287dcde4628e181cd0a422e39c618e7a4c1d18a5ce2ac12030a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "7a1dd72c0dde7367692864243b14400f37e78b71b4b1d29f5ad4e9da6e69d611", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "8ffbc14582c5e5fee5ac2c757bc2f22d90d26e8470d8fab0398028f1f99e3642", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "46bca6cec9d548a3f70c1eaf61d89e9b90c5898f1244c3c32722d11d0dc895ce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "d82c2d1950e47027823eba0a8c6af8a0c8a1bfd45759253a4e7ce319171be51b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "ecebafa9edcb88e6ea69569da94e629fe4237145f7a081bdbbde67b38cac591f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "83e1b96d7e662521c5222805cfab4ad3c96c70e23870ac6781f226ea4c74cbd8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "ad4664ff6b43ad535c4362b9a3544b1f7d7ab6160a4f50835124945eeb6b8dae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "02834a781548f1ac35dc509b6d35921f081ab7242ed35aba5119a6937ff65b9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "6cb67af774bc97a6ba1b50d49651cb17aa7ce9af7e39872e8a65fe95f799cdde", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "d256ef8ace1b960c20a8e949365893976da900a9e70891de4940d557ed4e3990", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "7a56d9a0eb3c4552a8dff64603c0a1d9be83526331aabd859c376489225ca8d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "d918d99abf17521a6ebabe32130c34ca7d5f702b65cc9ffe40b733e24af5fc15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "9d4a0b8fe876e542aeee7cb94ca7e2937946354753670e3c5338a301908bbbe9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "4748f6e3f0dcfc0ef98d6dbdbafde28ff0243910993e68cad29083f208c0b345", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "b617b1b5f4d0ca593b3898b7ffded78c95744cb77bc71bf0239869554486d6f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "d43236315f0d91fac7bef6c31166bb8e36371118b39e97260c3d3bbfd34a0bd5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "2db1bbb96f1670ca0324d8a4ce9ba5309450eb4f698aa0ab76b2ffbcc2f8bfea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "cb2d3b7d457018a7b0fd49ff20982a48a9e85d0acd9d9ab96d31731856a7b408", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "a2370f1c9f13af2f37239951c8fcfc1413eb813e1a4da0a0ad19aa5bb1c69243", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "7b362d1ffdf9a8b20cb7029986570017a68016f3fc2196461f449bc422edf2c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "7473ce8148b11ed20c6cc65986c196fda5f18e6acb6d41119d90a674725683d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "ad5dae47fbf25c573e877ae94d168a489ed4c5afba6c79bdf410f737473f1ad0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "2fff56127d1e8e765115ed594c8b6cdf8716f5ba079de75326648dd9548b1509", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "b9e2f361d982b62dcfaad5e124c3a07b7c22ca1bdc80d6339426d22c454c8231", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "8f7f8dd360e889084bfac5c9f3286bea60648726a82aac49a33bbca60bd383f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "0f3d670341cbc8a63f4d7693fef5a86dc170ee07a250c7c8f55ea8e9fcfc793e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "164d4d39b121ab7dfcdc8174a7703d3d9ab7d039f16e9265ccfa65e78f0cc1f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "a808477228fc169c08442b500f40fe46f7dd020e854474b76378ceaf519d8e2a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "372ec53b64e034f119240154204d11ea7476398084891273093a8988dc5ba19a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "4460abb6552a58b018838dc6f61803a435681349f4681dea8d5fcae9c480c6b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "71723a279c8d91d45f998751f640c570c111292a662c25b738ce0fc7be9322af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "df5393268bb730766eb3e1ed06a0668703ff0c5aba24256894ae0ae6902f9b05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "691d42e1c254d98d0298bb255326dd3b84e13fa5000f26f434ea5f59b7a84c5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "26bd4cf81eb109921b704d42860ed8d0251b8e8996e6982e154371b0245f6cc3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "48ea401d7cad6313da36c2b96305c9b49cba337c948601a72943dad6710cef93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "abf823e614ac8095ed8efdb811a5395eaad781c5b786eaf30dba1941b08c5cf0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "7473dcc187d42939273e172237f0af4b976df517b136ac977eca600536a82405", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "eafbcf0bc65322b00bfd5edbd5181d1b09178d62784c478ebaef60055af7cdc7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "5ec520fd70cf4a8073d8a0f0d0468c5eba19ec25f7350cf93a6cacfd0b5d4f5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "d62bf97f284198270333e5f900fb4bbc635f85240d013472014da379ca141fb2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "6ed45d5d41d55d9489e2539b12194c2efe4907613f12441eef8b8496379f537f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "5c09994216973251d1048db170b6c643f2ca5ac0bd9c916aea38d4f926fbb2af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "ede86d5520bfa39629530234ac91a2aaaaefb1695d7ebf54487503510ec85da7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "b7f7b43e40964959e730c15b790571fb4a3cb27a29d90bc2e3d369e3922f2623", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "6a6bb426c81406a3553f1029b3c47723cf844dc55a470b3d394727ca42ec1e12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "a5dd1c9bbe5ffc8bbaadc2f5e11dad07f5c1fe961e0e525d53f8642238a5d99a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "2ea03f65718bebf4d166e33e1e93157864173d97ef0ff290efc3e0879ba26ab1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "c30e54a8e59dc1c560985f8b0a4c99f5da132764fa97abc785ffd7461212366b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "8a947a62d6e4241c90ceed14c327f64730e1e2a90ef7bf0f1beac59e79af0186", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "dc6cc5f7537f8b2b1e0bb35e3f388d83c6714ccc88e03c9d99f776b2118c8577", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "6656b73292cacfb323ea96603fbe5eb232ae7a68b8a8db81e1cf7a0a680e5165", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "c58114fb6622d55722f354df40adb137499176b4bcf2892e45fc490a1c54dc12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "d91c9645f8620b58f09a00b79ac460fbcd654d196fdf5e4229242760c9b9b3a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "36291a3650f50543779ed81e389a6587577dd032997b4a36cd8847f872e635ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "782365c22676b26655a9f6aacb20cab8d40bbce2b2bc458dc8ad5d9646b504a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "b50a2d33a36b9302b5345fec675dde469bc8fca7c35fb82e9db70bc8396c7f91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "3b3050153b603105bfae313c168a413911c9158b429f545495940762149a410d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "5f0695a6131c3cddfeb8d6fe18494ce3d5b561de28ff3e5401cc838198eb013b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "270ca24c97cd1f52212db627231cde3d26e4b5d755d386060383c5ad66d74078", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "0f2c49723bf0de548d7de6851a1c9d12132b7dd28c844e6363d967c5af9a5eb2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "286d6775d3038a7ab5639de0bb5bbc4af090422ae9cd3ac431ae07a93b693c92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "07464b0ae8f7fb5959671f48974dd6256c7c7179d84e9d11b710f38211d5f954", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "9a2b1b1c7c74caf455eac3268a73f31c5bb62821b858c564dcc008b620719931", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "ec2e41ba7aa4309cc00550a0dbf23817a31c9c20e5e4b9b90bb19f302ba7a8c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "77bfe6f8ac6f44c258dbdff6e9d463aef53ba3da3a1ac267e4858236601529d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "64d4033c731b9eb9b7b410efb62056cca4ec38ca30be2ae2e8606745a1ea2832", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "5edafa6e5e2b30c0164b012549573efdaf91ff722c39644078579004166cee58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "141ec2d7ea5e2a9609b8ee0583eea102ae6a3cfc4975c49e8362d8e626573032", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "0ca338231b67edaf6787fb237bc62fb577d1f41206d89f7ae7b58818343e8811", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "afa3bd0ff9e6be8dcafbf2665f66baaf201e0f33a56b26327514cef8a07b11b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "d7a778fd270f18278718af5483064dfa1c0989bafdddc6cb1afcd2363a1eeefe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "b64d828845bb00367d04059ec7bd6b88a3705b9469cb0281e67a8a6c6d8d1f42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "68c18ab8828f0790bf3f745ab1b29b01a605f67e57c639d8e6e0b4ea4220dd01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "a7cb4da2be66336c2e8c87a823a013cbd4dda85802188e132c8671543b9f555f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "24b34850200c0eac04a68ac22ad18401cae9bb2ce6c3cdea7d1607aee79c13d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "8c5c007e18a24b579995cb7e8e448a6d91b27532f0cd22495c9500baeffcc70e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "bcadcefec7d2ae63fe6825cb4d2449f3a6c051898970156a9efd2dcd30e03cdd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "78e078bd59f210dc2a23bd9d81096797f9f97b0ed78f4abb2e02740c01aefda1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "246e43fd45c079a3a326a363b44209200e4043607162d8cf7df07aefc1a4dc2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "6bc31e560e4bfeb54531d35bf239a30181e3fb7c8a147e0f59b9b0108f21460e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "6fd89fadac875c117f85f194aa711b3e767e101fa2f0e31bd5ff2ed1b6dbb7e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "f8d9eb774bd757968791454ade00fd12a044f76242f72c2420b501502f20f1a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "525f83c39fda88f6cd578538f87f01f3fbca4ae5e83aee2ded8237dd420f9762", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "fa7dc74d21ec2933b105ee6151ae3fecae0daef6c0e511b8846a5513f21b68ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "84956b311ec29f41613273b566717efeef90040e157c514c3f3766e2774ee404", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "e369a5bf4441602bba16641d4e17c598156e58f1afa427ce5f206917734109f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "bae41cb41103b52d623022836364a20706aaf50af53a25de0b64b1ed0311f164", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "059929a13907fe3aa9c5950c7e9c5c14d4a69d0c466454b44e37627733b31ed7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "8aadb04898ded34ee7c958d0fc67a02c50ec84603e834544afebd4a7635ee5cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "53d879eae8b22ce51b0e7aa7efb48796ae9f4195eb9dc2af5bc3269b619ea0c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "9a4ba4283c3164d6397ef97c71971d11eb4e67c441c78f8fa92d6a7f1ad9c7ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "2bef095e6bdbae1e378890bbefc24b0b1142ab67c537d91475f47a3900c448f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "892ad2184aaf0f7cff59c7f4b7106b5195609cc2da82d2b9d665a6f75ba9d18a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "bf599d1bf2775a21b6e1cacce3f6c657034fefb7122485188c9a613d9c656a4a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "f0473be50000e14bf6d7042891428d1f5381e4e859568b93b5487dd5c3df0efa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "4b4ea209497413f34020562724f24a32a58fb52ac422352a491ce3a743e1b2ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "4a66fe7e8a607aa5757ce417f9675fd6de29f0bb7a5df5167277ca0ed074efdf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "54215a8c2dd2780fccea759a6b74c9def06c96c444a1fb8f1305c523772d6ff2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "98029938dbddd5eca4319e3e23ead86b60a2e4ab8c598215adb08fb9b8bfd065", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "dc4fde4e7d76f5cb5fd70ba963451d86bc16d17226306c898e171baf78b86730", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "63e85ede4aa9d73421286527642cf8e8330e37aafc1e00b964a0adb288c11c43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "257634189c380225cee51269277cc48c5605f536e9583246fd4ef1b8f1cc25c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "4667e0e9b09ded86b6a2b312acf59f5e7856ea0a0692ee81645e7b7faa542c97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "c030074dd622eeab4f89ad3c646d20773090e6cdd0971f3764f92740781419ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "0fb70bd9667ef8f6951ddd41eb7662d92f2b7b16ded30ee2e8dee43917b8fe6f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "2998c10aa69673f598b4c95a5bc6fb53603de243d2ffebb5a0eff91ff35c6683", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "0dc4d70331356c850a3a1ffc897445a5f7dc7881f91ab4f3e8b68bcd6ab17b2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "7f024a231dbbee7474d11c21742e8b0543a71cb85630201ca95176f5675dda5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "df1b9c0db10d9e59ae88190ac5a54f76b19e6155626be27a7fa4d9aba549ad41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "34214e61270f1853574e7843c2da6a4dd3a19982ff9bf2137dcbe16d0944a61d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "143336a0bbd9b94d90929103658431956fc92df1d003a8c7470ec644ef3bc899", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "b55157b35df4985f62d13729baae351bb5886815d9ab916f5750409d90b49cdb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "9c28fba8d95afa4b67bd72405658adad84f8d766269efa295971b44e69b9300d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "176a43527b068b090223dafd17a054d81cc4ba2bd03f8bb0c3372510c04d2192", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "aa2bb18c660ea847b97ba3e23c728cb608f636ccddad64a86458034f0707494f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "b8a9c68a1ad71ff57902eeed970fee7d4ad3307619a90acac215afdfe184f6f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "330929d952a44a93213fff55fc67f1e33f36281c83a128ace9f8302937035067", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "efb09445e66e91eb70f049af6d5989ad4f5076ff401f6c0141beabb749baf6cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "2c2426b191b7d26c411dd4991fd9bbc351f48ffea8e9b3481b0b0e24f134b951", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "8ec7c6e53969b27aaaaa34d426d1ce1b83d79812e659ad78cc678c3eeac9d14a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "da58209e9dbe1f7b2cc772fcab12249d292564d6766214342b2dffc4ab6fe072", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "bc88c44e1180fc8f1a241a781cad7680581958c607b16ec069790f69084ed1e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "5363a6b8c88414959ba94f7f583908952ce1244fb23f1a5a79d862a542e7810e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "cd0a1d1c42c59078b17b1ce0761deb33d5cada5157814d3433bb830d8a6208cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "87afd92d31ea986dfdf73c9cc9c22a91bc87656ccc9819c5606db6c43ecac528", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "33f74978722d7d17d55c4d642e32d349bdad0093ae62dbecaa6997697288f870", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "f813254b73e02670501b53f30e11ef7b6a618bba755b72e95e8a7f45bb053521", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "76dd2323dc75c2d648b1e4629e5706dd1d4f7afe7e5b4a22535989cb73880f73", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "e8cc71afee110ca51af12fd86cae63806ce45c7f57a0ac19f51484da8202d212", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "6793f4f58077480c2dc907cf1fc0daf2a9df8d442dd5fcf0994462f62ae0d445", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "2b817dd580b91f9d2aff19adeef7dd72941584904077f8c47237b8d01d33dca7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "612a7f9a534c99143a61cf7c6267f406029d7ec7c80a2d2532d52debca42917f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "0daecb62e9e87265c46fc1feaec719bfa0456bff7d20fa7cc74b312992ed756d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "4e9db9f7d2bfd306ef3ea22b0d6a8715de490a35cec91a6de49e35e9b339170e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "ea65f0b120f9f5280529205282b6f8ecbbb95a19646d7dfe079673bec9b2b445", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "3c0bdd818c67f6741f4534c6f7d358f9b6aa3e833f3f01353ebffcba7c8cd47e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "b9c7d66c3f8b438d8f0290a5657fa67d963123543dfbdd435b8f7c9f09991279", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "f36781b61195a661c11458fbfbe274edbc5b3383e898e235c7906c4f0f686652", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e33a5e267d8ab40148cd9dd848e4931e9f5f08c44c00ffc99bcb3f0bedf38ccf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "1fcf658a1b5c59c381d943ab256b9854a597966ee1d6a1a14776e75fe1b8293b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c4bc1879b62dce9fc62332ea774f66ad51a97508632fb87c8279aacc9a77d89a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "88b510a172b33de1b4224aa2636f71e2ed30f3b9728f7f627482d2c8ea91823e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "833fee2a8205b0c7183a4a306f3b1035d9700755a225f1ae4c970813f5e9ded3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "60eb8f63cced1be816f1de3231756f7204959da1e402c9b295d093e8d9b61340", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7c954a4de6add41b8fd89c824fe19944480f641d36c1085cdc8ea203cc40b075", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "2e836e760aad9fdd13e9f2d4d8e909eef3305c40215772131ac09c656b165bfb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e41a4df41a8b6bba726d15cb02217fdc67a248ab17e5ddf73b481bc5498ffe95", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "a0e61a6c6c2c3830255456180faa81a1e45a2a4e649b284f4778b2a17a6d386f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "329c652c4eca2afa8dca6bf2bf9b55ba288e1ca58e14ec9e1f6d2cefaf0efee3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "d1f87f3a957860617762b2625b61e0619791efb10722bad2ed55325f2310fc49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "5ad5122857a6539e1b52c7cc3356b6c60de4515c443b39595579042bad43e42b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "445e959a416c70e69c56fb0fe13e814ee80a6b544e3fe8f2b3b3686e6b29bea2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "1bd59311b4332fc7be42348e75a1fe024c95cc34effc25763a38e8d06c764083", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "44717d7500d6900d9ae9f8f505351a41a3e4d0a0bd68471ed003c378fc3afe46", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "da66e775f5f41dc7716364b226bf2f8e602108db6bed9e3b143e2daa9959c152", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a9fd6829f6687e7f79c5e14cade87ebafeee581866b7c52d84a1b4c84c1af466", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "de44d5b0915b063062fbdd75d32c078e753ef53ea933e849e54bb74c818b8565", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "68dc8c9a9fabef1e6036a0733cf4cce876a0771cc76a198f8428208586949736", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "14a6fcbf01764d1a8b8cfec409e3a71d553177af1c8a7d47b0fbfd2e50c62494", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "1986bc9cec35081ab6747e2049e75aa921aa6693340f04f15d09a90a4257b715", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "6c7473a49f01608e28a369338ebe1fb3f533ae7f81d59a86e3e2c02962967fc5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "ba8e819e175ec37b5ad1ec21825f80d96672ac06929f274fd28db4a75643079b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "7215a732bae15d330ec4a0e273e208cd72429caa9338a7e33f8dede2573321dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "8835aa1eef3b0a5b2a3ba089053f7b4dc1a1da3caaaf2bbbbc5f2c2de39a9636", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "2836b5f48480407102e226018e01cd9c96625c4bd7966951e251f7789f4a9c91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "3e3bc95261a5e6b97c077732aa7e3c58ec4090376c4235c2b4549c39966f0f06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "0371e249da62ac7fb5e504832894385a2766735eb2ce494b40cfd33158e6eca4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "ad5617becf1df73db61247cedfe608ac4fb26714459fa2dca02b392833945845", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "60155e18be151f76f66b75efcf00cd70acf336d6faaa8ad67c4dc6073426d54c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "1f4807c1e80f3ee39d37ba42cbff9567b389779de07cf2056369c5eac686f6a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "954c5585c9eb9e4b0526ca5de8556796e1cd92f5fb935bc12aabb6aa0825c71b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "9944ef3448b212b0e53f68abb10a42de26c3f3e558a881a6d46fa4fa6eb64a76", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "670f4cb0fb91553e5d1f63f3385c6edfd9e8cff197856dfab137e2a8f1e88161", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ed65e5272948918685ca73baca1ae614a3c1545ff26b93292ec13e0a3d5e40ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "32feb8b24a5ce9b8ea43aff2f46f6580409c5727dcd8d85b574433291bd99811", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "282389656d0cca17ffa1a6b70ec8abe6647bff58fa9d3cd5a7b3f87ffb29baa3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d5e698106c948d5f5b4d1c33722cdbdcb9a0b4881db920559fb09d193659fe8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "80173754e038a4a2e19dc9a290e225b3d6a2f85bbc925147a667d1da9b10acde", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "3ccc2416a1e1650e6420ccfbc3aaf1dde6aefdf5da6ce8e17d9996253ec61083", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "82991b6c73c90cb28064761a84c27cde642441b7eb9a4a8b245cebbcb0cbe023", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "10d93a0bd418cd2995a36eb5808e30374deb2cf25bdc0b492e506cd4d038eff3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "38f8f01b57011469dc5e8aec4bf2b5af0fa316d1b39150e885de238e7fffbd42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "729bc3fff5bbe6ca4357ac91e9060eaa616bd40f4620eb9690122626fb63c23f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "59e15d95dca3f0ccd9751f5020f1c1bea58cdfc56d2bae133f617120e4644d5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "6eb9e14519a4d99e6b634cc5aee0dbc92f7ef4fdec173cdd216db5b0f98868e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "738b9b493f4d0542975a9ee624b431a90ed1b63f3bc1ed330f8f79ac9b86bf6b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "5552317f0af12c57205ee0254d194c671d2374b7e6272d248ac9f8714fd75e9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c3b369baacb7ec7dab8d424922abacdec8d75a098ae328db8f1229c86cf4d1a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "2ce3da589a5e04837cc9ca3dd4800f4fa2923055ef6f1fbaea63a65906b95bd1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d9c153e74c234829c4fcb680fb34a148f19622fdc3de6a916809193b50ca1685", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "2165e2112460793e7a1152dee32c682a2c1eb94b7ed2087d81f9d78de817248d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "eeef0974edccc99c1718130c811819c8479e5a5ba90091e59c814b41370ec712", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a32957ff507af6bf3068867bd482074e25a5ee58168ff330b22d328ecf35c7a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "4a1e2b110cb7c30768358609f222526bb2b0b1c7d293f95a6a8a4445ac7a8ba7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "f9fa4f8c133734bd9332caa47a5a8d1d76cd7e8bfd3e34db7b42eabe7a42306b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "986f086d746f528371a9d5e31cfb8db6859a8494ad1f947c2ebe2b03e87056db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "403256323a88a47828b8bcdbbda1f3b811863e51ed04be8acff9cfe66955ba56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "f46129a8ea17e5bc5d3cdd10ff66c50935baf83bd2de8c983e43b93589f8089a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "c3bb61cbe8e98aad0b7a922097611333667d2ad977887024fd70e0b41ef8470e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "7fe06c8144113abaaaa495d4e4c046355245cc3dfdaeb6724bf4dcc3d19da1f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "5cd69fc9b9d71ad4ce8d5bb50b4fada3a6604cc14483d6b0c7a6d0ef377240c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "55b461ee332e425d95ea2152a73725b519322d00e0bcd2bc2f1450fa46645b3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "8de12079b5963c33ff2c5aaa90cfedd33ffaf1e0d6da9dd76ba28e2281404a87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "bb7ccc5eac6c538f192a321c86e837e2f7418e78d75c5a5151d0742fa7f53381", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "44b55814e2074408bc10e0a6927cff9c0789654d49cf7d0e6b8d7a0489e4ec8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "5f5142e7322f6478d046c50bd03dc03657a2ad229f88e8271300d38002daa3c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "438bca607ab0ccfdd87aae95edf8f585dca8a28004712a55a3b838be549b2fa5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "0a42fa41b8d8afa1e41ff5aa4a5bb3418d07100b6596ab14b8b9e8ec9eb1f091", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "cb928ffeb29b012953cd8e97ff1a66d7dd018096802f7e35d5d8c5fb23d38354", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "b8c91dfb0e5cd7ea87d7dc872a4366e2ddb54a07ff341471e913e6e32c0fa237", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "243f5988e3b543deeabb5d87a181a2df621976fda16302893c42838b13461beb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "f562f7ba2606d2fbd0ff1090f50da7366884439be7de5b442af7800074c430f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "1c8ce2e19ef1a6a63dce05eda89e0af16d9d1cb7ad9c508343c5e19b37a6cc4f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "271447f16c6062eeeae9605c22fd7c71829b2bbe25713f39eb19003911a4a06d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "0fa10ab51311901b3a3cdd91f257eb30932fb1739c001bc2b89b253fcd47d585", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "9214adf45a04543818c48dd2c484a2c1f173ce17faf94a4fd08b161f7df7a488", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "72da81e21ffc326312d8c65b997fd53ff10cae420a67d1893e2a7c698e14c814", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "402446240d987580a0c39613c739b7323756cda538964e4b3380413500054a71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "2c1d3d2767c8968ea833eeb8493a548e803f4a33661e810d3481be3cd06b4ead", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "8ddef48c43275ebab26e1cba79b2b9f49b3f78ea998c44664cdd6ce3980c4811", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "ac4ab15958e3f0bb1ca2aa0eedb4a40518f0bcd130093f49135c48feef7082f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e614e8eeab56e6d8b50641e289433df7dbaac058a9bf340675b2408a2f9efbd8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "791518d257579caad915fd95058ae61aa3bc8e9c73f663dda87d4ed680271002", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7749c8f3288d3d91aa987e213b4fec4d85031aaf323798d60a9120a08534f31a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "723f04f0b93f0d12333abcac6b10905cbf04a881f02f7d6f8a6ff76e8c441ee8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "019c782af9080dda4691f2211988ad26c59ac9d5bf89e612ea3013d7f5baab2e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "3c3fca4e9d8cf738b1f548a4051ada380ee6e5e141afec0f03d526c9dd79b595", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "22e4796a9ab44773ec3eadab81b2ab4473b22a3cbb4c6bf98d6183ff31485209", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "654130d8d9b22f6e97360efa63751bc105606ddf0d8cf5f695ec67d6c0009781", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4d1b5123f28cd7bbe62c41862baebd4eed9d75bba9e6d498b51610a7e3f017a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c28b28c723570fc7af01067a0ca70d9254c9d98e9c24ef2138a88cdec728760a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "95b1c82d04e7e5ccf9cb0baf325c14d35b837fc0eed0ce950773e7d5001c0b76", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a46120ead440c029ff554a850e9fd69f519a0b9e827ceb7e757bb725b4d56ad4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "5c9e617f56c6a01b7ce36be4ed526e7ab8d103db841c67063192edae0bc8d905", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "09a7eeed35671d56e4859b062967645037e0e15d31f6999417154c6c7478ba2f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7edf07485a42d2ebc3acb4c92f94e69de8fa73f45c1fbb42b38e5ca3ce50b845", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3c4a72e2372ce6b9a50c0c374404a314fe7c2987ef189f46ded578a81ed25a3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a4e15e9c5020e93183aba6d945438b8491da9f70aa20c340355ba846b65f61fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "94dae68ee643c282d24d44ac318b863efc498014e6b3db39a283bd89cb3eebd9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "bacb3e8b123b42da5b077e2bc04ce71afceeb8655043bb7c8994b8fb97d0e783", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "02618a2df5c03f5b4f700b53f31b5d44dc838172b8d327702a63b2fdc4545a01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "9dec406b09fe95898c83a370eaac38a7bf3e0a7aa468cfc0f049075260e84f10", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "6236f7899fe9ab6f33aed1a92a93cf43a31fe177cccc47bcccd87a59f6b01583", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "be7c5a6ead12c8c028e12e969d99dba3a1b7cae7aa24ae448692f2dea209369d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7743a57385d8579666168862d3e426a4c6467e7cb69f3b35ea15ce16f2e45f26", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "422f6abc754393ae32b5bc8af5fef078d9898b2b290eca9ad316dba9f0fb6d25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "aca7cc7c702d0cda4215728435b7da2547badcd9affafcf2cb6df8383b6316b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "068fe1671f817b2a6e7f3d5d0e8892d4eb8069742e5e9afbbbafdd03715058c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "61f21e467b73d551f9d6fd524cca4c3f7cf2361c175437e5f3238de0c81be141", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2ce2ef563598c5db3118429ccc3274bc3e2cf6f075e71691f8f8f08f713b9d96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c1dffa3ac3de957ce38c71ff06f6b85c804b8b65fe5ed4e0b975d1669beab08e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "fec3f6183ae5c99dc8fcfd98101e3bbacddbe6a95845df413c74e7b5e6eae542", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "391c7a0c3a3f1898e75c060e5c607c0fc486e5b2259bd7227b29b140143078fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "9882826f706ee6fad19cad7b6b0f94f5a964e45c498df29de9948e8b0c52c173", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bf474d53412cee9d41ad96c7cf9d060a5d31b2e69a3f712c72d04b49b4f5890d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "070a64c013c4f65020ef2468be95310ebb686bf2e40742e0674ea16ddb561d25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "ce749fe816650a18824574265c34f95b4b67be9ebac8d9c686b3112e2037a269", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "95082f364bcbf3d86e5095d5b772a3b34b4aa7281f321197880c3ef86b7a8d46", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "5dcca9e3a4043d5c6f9a036f527dcb5b4b8c34a8a15e7b9e1fe13121c2e6a96b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0f7d9694f2bb1cda3d4ddb8a2fa1cef0a35b7d0f0ad3712e4bcaa08cbe54cc40", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "baf0b6f3e6131421d98b03a1a3531bc8bec15ef66d38d1c924adb6d548e9b732", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "26a7cfdd9eeb20d8903e259548e7cedf120d343c1fa76d080862d977e52ab241", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "56a41979b6e0ac9ec886e73cdb7bee6e0dc205ab2743553850b3f66f1efd6fe9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "a20d7753043a330f0f4ec6a097371a2908cc826da1cb5e1056ae63e2664e6cc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f3c89602051f99f7c3b1fd1dbe3a19b994bf26167320317cd0968d8820c36d68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "27a1fa61ed529e62eab2641d01d890851996a49518d3d0b5b4d75202e2647872", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1aeea79abe88c0399867dba462bb95efe4d92d0cd642215a09243c8c19510001", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b37a08a668225dba7e27d40aa12263f2ad30844c3bdbe797c450c49522028918", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7c6f8e2884440316e2d8b6f7186178ee2ea7dc0cb87ebcd3785c307342b0dbec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1d2e68f29b066326402a277b6dcbf6257d892b83a6a2ab7c0bc7051f9c7749e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b395ccecf96094d4180ae72376c80ecafefc988d0ee43a3636b76450e09d273f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "751bf8b3d399e706adf0871f9e5b5784655959915bb7b326caeaac5bee2de3db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d369d65ef7a8732d5f2bd91a18327520c3ffdad6443a7565dbc5434f413f1e43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d948c4e417d1d4ca62bcb1df5ab92d0abddc083b2410717e0496568a3a003b8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "deb6d38e7c72ffda22894422ee58763a2d9ff81cf6f2dc06d76fa2d9c6f54436", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c3eea0e5562586af5f5d38037ffe7cada9dc5664ec9ede2039b5153e6d0383c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ff839e4e21fc1f8d82f0142dd41c01a59443864dd93856f11aaeda0368e2913f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bff123a0a9f2ee32b6eb33c58bb0b14cb4ae351765d804e41186ea17cf98d157", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "71d2b39a9a6c6c7f880632c2b722fdfa1ee2459b92bc246660e8f53df74276dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6dc1511a1c00dc8dfe693f73b25fd845f7ff44ee80418b64941655d444a0e1d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9cf307bc2d36a2dc0a501d309cfa312a32f8d5706b99d11ba577aeb8232f1f36", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4b016795198b5163be7a319fcab08508234b2ade50ad5ede54146a7a35bebd97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "40fd6da4c38daaa0aaa8eef81c76ada7452b7fe41fa7ccddd2d3757cf4e4ef7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c1e87290e96d85df58964af915b88b4680ce2d843e9d788c139b932b5113237c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f3f6cf8221fd7177664862ff379c42a82dc441da3f5a0e13a968204d4692d21c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b67b046d5e851ff42f70ad38fff3df32d6f83865f33a62cae435189c677a8e13", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "718f45cbf93bcbe2b428c501b921456548a7bc1f25a3bde7183a2a2a72106d00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "3dc41e6a376517d2905f025315d323ae04b137887d4a91a52cba6d5e930f9bc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "f449bb1b1d56972aabf7933d4714360fe93ddde21c03356afe32e74090d60edb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "77c491014b9e02f7f72656a52c70db29860586df81739f4e0684dda1f95362ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "2584aeaf80e92c9176193d3d24328cdad341f1852583faa3db62c2d039babc1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "1160810de63802958d45acb2b25a349c2efc6fb738382d872c48f5f410f3293b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f8219b4ede350dad60654792b0f56a2c928e42210585cbc4c5df94e77a5c8171", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "11921917f37bf896df83ece3340a843cc3ec69963fbc36e8b923db3a3455f8e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4eb9bb187b422285bd74fc1c9dcdb6ed19a5b98d6d50329124cc143580a987c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a6a826d43ebb06e28032dd3ee552c46ed187d964fef91e5780531f1b31428aa7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "16f0e4de3001392abd3191614ab4dc8dd1b53fdb2dfef78b4aa7a4455430c73c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "83d0a34281236dd989e1c777a505551620ae68e7608c831e8e2f0eaabf95be77", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "ad4e6eda01c44b629db831924e5b4bab588c170c71cdc4019d296a77ef7707c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "137539258070a3dde14b04a986b1599c780c98c916c1d93a620773bf5842dfca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "fc2a5565d64b334d0aafb75ae1166e2859c364e7a277865d754c4f1a5260a5ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0f00372d34f09893d3604e431f9a04c01d935f9bb51db2c50c0b94541f13399a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "ddd21a639681e0b3ba1e7edc3c54d6775c3011f0356c0f82c00cae5360643229", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "58ff773fad84798eb65124a7e6bd91ba8ff7f318af47a92c7cca6428c63e1226", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "dc8c3fb2274a487d6d6c3e695efc46cb61315829561a2ec092193ecf9db2ef25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "dfd523339fd512ebc4b8e0ce74f8d1d2df82b06a30239383963e5f4c7c0d12ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "633f8742141769b9f55164cc0b54f6c9a220b25185b502280d505926e8ef86b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "3f784d3b04f056406d6ba0ca7f5f040748b53cb876f9434c32cbe5ab518e1917", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "29cbc25b8582eae0f13c862b448ffdbbd279e04346ad41a90eef1922a349da6d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "6f9e89eb0170d4ba863b31b06ee030949c714d62d019e1c815be285cec2beb01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "1130e5688d23537aa69fcbaad44651349da01a728baaf9ace4086bd8b0417a50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d0d9e819c82b507853fd336689223ed7ad8a305efa01bc9cea34696969bdd8aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "31d0161fe2b918b5535021d54ec9ccd87b4d45904ee3ff2e93152cfeee245c24", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3aff5689b611cdb224e6c5efc90d322a07713a4a691ed4bfee1313cbb710a26f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "72191cb2769bcc67fd871d6513ecd11e283c0cf12d625940f63f8e1f9ac9c05e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4cbf10d728836aee264450b2e9ebca2fd73f69d0afa0843984066d8fd502172f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d2f7f0653d9aafcccae323ab37994c32c8669f12870c4b9d2e3bff8cda2a57f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3f9419479fa57256d6ba7b391f5aaa822a5bc9f45846910996bbc560c1a3ee98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "28900d1376780464ea7137f9bb6e22ba469a6ab890f32e614c0d30742935a159", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "760013b565c00c2774b04aad30cdea78010639b64d00eb731466644d793f4a33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b68d79d0880fa83fbf353f6f8c492c384ad2350d12d77c9bee879363bd2067e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "c58b7131724847997fb55bcca6701840cb28bb62c56255d4bd670363c504e4f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "fb8d8724ff2ad22eb96d1903bb7d4064f2f599c781e25edd7a0d1d2b2e7d1295", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "48dd76275fdfb43c7380992ae0341ce91cdc30c0b977317bd60507f4e5f10aa3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "54802d1827fe6f235727b781ae8aa00ac8d38e072a1b120a1045e7f8886d441f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "71dd6c0c1fbadf2a8239eed431f2a0dd0122d8d70b7aba214a42983e01a31b28", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "1f72ba967197527871fc8b9f57725220a5926885e2cc5ee7248f9cac21a491b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "0af9ea638e6ccc62af82847a3864554a6bec2f4cef371e0793e71a38f6123b95", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "53ade25ce872a77ab51bd5112a7c44133f3d608b1a8a38083be5215d0d39815c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "b3b0a4a358c80b1cd86d013d759f4dc0ab065f203931859d7847a732a397b521", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "5997e41a21aa0b23d94e4e1aae50ec996598708d0491521b9ab86945af6407d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "68d298e2506979f751e05434a7200b85b80277eaf651e5bc6ffb0f06e37a0c50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "16137e0a3f7772fb15b89316654f40a8799feb0636adecfbbfa340fb3ab5a162", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "a7a7a5f60205b175675c5c0a23b3f8fa3acfc52036c2eec3519433a6baa118f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "239955154b540a275ac921cbed86a516779c75629160f9d6ca5745e476ddbb59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "54c12a2fd68b9e412ebfe487580cd4859bdfd70075f89edeb8d7ddca30a16de2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "1dbc505405615790dda5e3aeb699d12f49124c18f98cc7b0962ef1e201e6ab2e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "fd150427b33f9e5371ae13a4acf014a8bbdc45fb995becd01dd04f74ecffa4e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "2fe50d31ed99a26994afc1ff11aa2f81d5e7f06d476c2f39204396e6e6830d83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "5b0cac52fcf62ab3a5063e88190866ca155bf1c70e3108a9b8f6470b2e03d35b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "f846abd2506ca401ecc537d0cfcc6cc8ecfb98a683627fcfecf529d392ae9c53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "13124637ff8efd49546e48e81949f3a2e164645dff23624989a338750f64c056", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "c2cf147492fda17644886ff34c6dc2d0d224d5f936f959d34aebf78ebecfbbcc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "d286739392b7b69a0ac258db8e317f1ba02fb8bc168e6bfe99baae4979cd5be1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "7a98330cd1eee4a0a9a26f31826afaa24d9a593bea61875f76ece8c54988c649", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "968628a6740b793188e6614b0e311a740cab59612d824b88ad0b35713afaa563", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "779748a3381eaa38460d4dcbb36467f607a49620f8cc4d58fc667e0d3bdb9883", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "8684c1bd543a1f32fe6ce3c9f8ab4874eeaf5cbe8e99be2b9aced16060683361", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "c6f29a84e796a6f480f3ba51d7445cf586acbfb5391134d0fa30edb784ef507c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "f917412df9ed27c705f957ab22b07600244369eb01bb565455e7f2b7272cdf74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "f494d1d5925f34058f13e7e38c1e3359e3b9ad5bb82bbb5d92cdafd80882692a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "d6810da2a82b5531367207237a5dc079aad1ca55df086ee50c9fe5d86224e1ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "c9df66c165d07540e7ccf950c1929bbcc91c2ae4f10054881e44dec402b27409", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "9dbb455c58697ae257898c8dff43026d044b8f4a8551a7cc4b8b0f5db4ba3ecc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "e2438ccf8f4af277bd3fab23c3a795119df33f8323f5a80e7c7dbcb87fc4e325", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "bd801f5802f1a83dcc9bc095ae2902dca363b379628d75a42444cbf7cbda61c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "9b1dda2bd2ac9e65ef532feecdc09398f712177091dcbe289b9a3eb1ef2835c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "2a9a51ab1777a0f7923dc5ae4d79e6e2f448dcf07ee4790d53d02ef46f30f925", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "02d6174b24a7c69b1343f5f512c51f719215c546c40776f7365c0165caf4e569", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "e71297a5b29769851c414ec72d447ba86168852cbd98eaa73fdf2efb546b9243", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "afea540f6c1c4c4b12e8e013a547b7dcad412ede155c1a96701bc43de9682f25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "1093cc6ee137fcd97b95ac8642de70600514ccdcef3140143e319c598ff359ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "e2faf197b21bdbe135e48c17614a409f34b798d4e35fa034c39bc811567b0fdd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "a4d1c29297988a9db132b89dec22ee13a4b0f49f13c0ae7c8b0c9b403c56f618", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "67a252a2e4ae34abe5875555951725854e008b6a196ab93f0ccefdd8e1383dd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "deffd233358a10c5819fdb0e8546b3eafbf98a74019baf49f29561c410daf1fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "166020cb88ee6e693ad559c6fd099c6c2ca784c1a9cdc05b621aec6a1defade8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "580cc2113f5dadf95ecc6f8037a67fb4d845d88c12d00808ad0f69e114ce631f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "456c4b5f77cb3d3d17761a6499f036eba483bc71d222eeeee3faa5d87e2277e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "55de1194d2efb8a35840972fe09d69f218c8f86a402528e662ec18f8894dd8ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "107dfdbb169cd3dc087b683bf895829c646d5fd5e39069ce4c1eb9b4554417b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "692a84f61e12c5358c0d5fa4027c38e1188c13b4e2d04d58bf9d3b930fbe9342", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "d311113826053117fce1e1bafe0df73d8a81f5cacf824350af55d370ad6731f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "d39870a504a11d00cac317aca9161ef1bc4439274acd0179ef3124f026c0a90d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "f833c186d6466af110a1c7ecbb7206d2a1f7abe5fe50eb66f59e753bb757e6b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "223617b871a3b2ab5c69bbff887c7e9d41554930d2d3e3f53172433a79216182", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "d126001c95a655574d0474b50016288cc3b8b48a21d4a866aa4aab5cc8ce2a54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "7710405580b5954a21e7e3cf0f79c5fb0ffe36718a239227360921ec9beed342", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "464635685fc8e0b3b0332e68267a23081eac9f512fbb932fe9d9f08bf489d2bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "a98ed56c1ad249338a1a602b03b378c9567d5567dd84c48d8298d14838c6ff76", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "9a8f440d7fb06dbd7eb3b8c3be5004a673e1a969e53aab188ade8456dac016c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "b2c9fbb3077c84049074928c4a4bdc5e14cad48e01543c3bfc38c4ef5ed7d089", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "54d5b00dbc691772735f3a4554c24876157124e6271b1ba3c2cd3479504b3c79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "b49065523a58819d90f9e6ffc0bf1dce34804a342e8a4271da8d442b82f84158", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "e56f026b3fa41a288d2ee355a8fd85725a5c959bd2c406fe9e6a803db52ec003", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "2a2aeca8ba7f9636142d7cf84c03f1fa067d333ce6e4fee820953eb152e1a9e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "6e1b6e74b022508b87c42b7f11f0fc5bca7898d9bc170dfa7818f1ac36585e0f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "a1ec7b21289fbf00658a2fbba0293cf99a1f09d0a281f512e9526a6fc82e02c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "5425613a951e56a6f016c09ad98cdeed268cddbfc3a4e9a5977f10f915295143", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "6d1dffbae1cfc6dc13d6ea133391615d97bd5f8b066cca9a5e3b33b90f474d29", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "844afbff45fbc333d4afce5314e3b5329cd7e1fa0accb99762cd6d06c4dd6630", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "f6c91fbe91ae6900b2bd9d5e82dd4e591aeef09888cf06ef7ae8d22997ba7fdc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "a3e27ae8db41296fd2f6e735af4970b0074edb98af976051093c3360869111d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "071cadcda0671f6be263550045bc58ca24674feaf0d14a6dddc7126b974c1159", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "2835ac2357491665727172f833969d226990680f3701cff41c7107c20795b1a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "a01faac292a4dc09afea3f26c8d634aaf9e8a8db450cfe92f367cc09678d5cd3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "ed4bc800d5a6c49adf007c2edeed5bfc1089fbf4809c1db256832461bf08194f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "506ffb035ff129a14631c3e902e515a287399bd6964df8af692e000f21cbf975", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "2d1014b5599c105332a29f1ed7a1945ee3de196fb6d369d526226b7b4f4e7830", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "7a011ab5b09babc7313c45d5213dc3dd8001af023d1c54a8d5c026947cc2effb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "8bfdefcb749a3bba611f2252b2649a4a9a8f707be189c9cc4175f443f81b4d07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "b68c345c811e0234fc372b15f4d86f1df36e8193a3cc04c2e1bb68cd8bdb27ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "93ab711d460b4ca9835d8a0711e1a820865c728c44e968b242fb7ef0f111bb80", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "3eb9e92b3de2ac60ffac4f4be22df416dbf1f8985a5c03fdd85399eb6ab7904b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "72215348d4e945505c160a34bde00eab5f2a50ed18fa69d9bfae042f201a08ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "420f381b17868ddf916e86ad9551dfa000db13999f0133db6db28fcfbff0f76e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "5962d0f86875f7ea95a30e6d51f2130dd7d7e65cae536bc4bc250600121b5ca9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "5e938b0d7fb578e7b935fc2d01b910fa9f303fee030128161bd3109a669c3cc7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "ff78eb7dce04e05fa0c84b6c6b45821b788eef3f35206b4889bf0e9bcab80779", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "7872bc0beb43b0ec20010485eaf87c851019607ca6bae021154a6b9455dfcd2a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "e24d12a499ebda5f937d4035a064dd1069eacc2d8a3e98feea8ea0ca96453c03", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "03dee907f5ac0acbdc7db998858fa88f21691a79148419424904e2a3c7970256", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "d53c2f4ba71eeabdadc6b5568c3a13d1b5d47fb9a2712c07744ca2ebfabe9351", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "96535eaabe6241762eeccc7909139ee85218cfb8f026bfc66dc684ed8ab5e38b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "decda48af86e1d6908230b1614b3594844c43fa231321d2e1eed9e5c8f163133", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "d752fe84f27f10012e4c9867b7792bdbcc4381f02c5978c2aedade6e84336320", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "1517035939036aeda8906dcbeeec128b373a5eca949892737e408adde02b006a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "4e6f84677411ba6ee0f9862f08c9deb6063aadccb309328e1c351c9c391c2a2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "2d9e008fb9f78334a53fdd685657a1d0c75efb142fb0f62fc352f6c6255ad1ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "c822836f049ccd8ad297a1293a8cb460d58669d409228b551a1b4bf00571498b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "f46f6c42743e1c53a8b8fa441cea07ad82f471709f9212684c101df2d4a23734", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "fbea27f2fee1ca533622c33e3fe5eeaa58d99a22b206dac4ee36f7e49a12dd1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "d75345694f8895a3f4d79ea0afd2ccd6aae75af5e9068442fee43b3bf317923b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "7cfb42e8846adce970fb0a09bcb84c3605e63fa24c4c37f1a9a98ea6ce7286b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "7786191260d416e74e13ae6a847cede56e61d896684ecc0d47a15a371d976fa5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "84d2b10e0630e1a083eb040e7582076f007231de98b267ab9d9ae3dcc270f13f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "22ea97a52cc7e9b85e5f271990bf6727e11a5a97bfb670249a242b5746a4cd15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "3c0e449c8e731b315769a76de7e1057b87231bfbd0ac563ba9e9f1759c6c5356", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "66869182fd7e5eec84e68bfcfe4f70cdf731bde3bf2127716e710519cb157951", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "679923a3d386164b79e6f1c9cfd8ac6dc75d321d41ff2579c3f97fb973232db8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "15cf3868e4e1a69f062949fc1210c9f10739e355092d50abceb843f2efa3bf5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "dc4954a05acabba21a865cf5e3a3ff5c6f2c8b11a720c5d9660ed6c70733cecb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "154a88a79e55fb08077216f7ab4c62976412179002c47780408ad0cfde45fd01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "ffc51945af3d838b998e0b1e3e0bdc7fb97b763002325e3787b36c227c31d4b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "cde3c29623dda53a2065701435e581cf97bdfebdd5751423466576347979573a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "c037b603484b2aec4974232606ace001485c16a9f7f6b520e26189e25de53252", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "e61aa8095a6129071abc7bcaa032bc57760508ff10eb776fd0e2a9c0a7d5bc67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "b604e147b3ee4ba16dd77fe96c5a3176c5c1452ba73a17af3b13e39de9884f66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "6a7d754cdef620f839c0da6d116755e5fee751aaf9db8038efa1a0690f8219f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "212ac9372c822cbd3700e6585f6476a8e258daa9f58b4d9970c20afe178067f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "b9063dd95e182ecd6e9effda91641e3a0d1713e511b2bed19084753356bd8e62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "7a0d4f9ea08aa4ff7ed59013e22edfd35de8b0bae0e2c5c9ec6036f44688ffbb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "c33bc24b222693a4d0c9aa52bd5bca6156dd9907450cea604f18eb72f16886de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "bb2a1b20244d10c3011b6977a82bfbe1c216e163b43268d3e9ea80f7aaaaad83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "aa4b864428355ffb2358397731018cd54443e02f3437539622b74f386a6bd35f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "60a635328de9ee02e2535f8093626e4d22242f22a2efd02c2073ffbd18becf7b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "0e7e0eca73675d0d21d9cda8b38f99b1849f9f5cf95e597b7abf0f15049f54eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "5be85c4ca2a3e74e66eeb9a74cd088a215a533a69dc88e5b92f6222e262996bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "f01da21b026b15720b381869d16e3310fbd9b2505460d8d2aadae4a3c7d49b3a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "5ae1dbf3edc6be50e6017c4e8fe028c143f5ef0f98955bd942b3b7e2db2780fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "862849fc270d4b98fc20ac58a46a2bd403f4f46dd28bd81301fdeae3158ad35d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "049a2c0df6165136ecbd36a418505fc10c590900b1e9b64692b508b3850d8350", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "38e4de8c546dc411d670462dbd7bd02d03f8565afeab9c3dfcdd4ee64194b1b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "f51476b95741fcc0fb1adc0e37cc56d73d7895fa935b26764ba46cf13e9d53fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "f6339c7b9e711a268ab296115c9f8b511ba4ab65a7082d4e1a3d50ac81e86644", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "53444ab6595cfbe82b12ab0fd5e130f7669dc92cdc034ebb901df32f00edbae9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "9493c18f07e3cd045cd3ff42fc3626fdff81dbb4c2257bebb24cca8afafc81fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "c93e1a9a32c85df62be7125c44d91c798be2ea75233c73462cecf0d648855292", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "b2ce11ed3f4a08bb279b2756b8f0bd1077d83294536925f6b25e2ab955fb9a7b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "0daab284e2d48c729931891369bec34de483ff8bc8336092f41201a45b89845a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "fab1227abea8dfadfc70d6d213a59a797de3aa712d57f32929ad625b2fa5055a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "467a0931b6dcc5f5b8c14b78863b67c5ffa4c8ac58fc6d689db400094dc338d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "e31a58c60b5bde579dc76322160e31a1501dd7afaa5d07b0811dbc08950b5cdd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "e2f3dfa3621529af78e95b9cd61ae45dc0fe5b5986843066e171244c47a32b79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "8bff225849021df2e98ca27c9dbd9fd242bec424809cd9ec7cc27c21c0f26632", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "b86cccd7e1015feb13e33be9b9728dc090f4a7aa4584591270d1ba11a15c5c4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "fa9cef625fefb2fcd8fa32e92e4435cdaaaade20f6347b7ee720f632e4173beb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "633e2c4a1d1b6cbef75787a8010d64e04688137778532f56e702ba3035305556", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "479b6b2a887a2db2d6d91ef400ad4651c2f6e6ecf1c6f019e4d706349a91c4a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "fb76113c92171ed5e9a0ad943f405c16e8b4af53a31b31d78e4c3833bf7b1f63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "3b695569567ead227ba6cd56866ab90e82e01e0ccb4639a815663b5a43274b4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "d83e1dfc7b0b9b24a20fe8c8f815b6e9cc2d0f1134f43fb09ba7d3f7243486c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "caaf5de948d53d5c77c06ce881150e880ab21e7c2b5199d4352d2a340e9eb4ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "b3c3f56445e711a79189f5aa9ebc2b6930c9082392893e98b71cf250e59243fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "ae739ac2806fd5e36ca004133ed11e7d10cd27b32bdaf50c072a605224f11481", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "526a68c9f73ad3e9552962dd70a41278e5ec38044516f1daa81fdc85a59e6069", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "a29bcfbf4588d759f39d3f412b2b3c96f27d6ea765bcbbb001081b31e7cb3d46", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "1d43ba7982385b4af4c5cdf52de174f0103af69f90ee6e5ce4d27ad4b11000a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "08a2aaae68193d0ab0b927cd08bf612010c7cdb359ba5595e3e1895012a37752", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "004a5a7b02694ceab987014b048a835823373e99688115ff1447004f2905a9b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "11a8850f3a866cdab7378b3453ab2ae07a83b6fd0d0e9b68d5b5b1f5e480f198", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "bfa0701d513ed44b8678e3cd79913a030c6431f31a88f26227107b477076d94d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "0fd91460ff5137283abb81f31c4f2525ec436abb6bcb7f1d0ebfcbab15f1b72f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "4b17f262451fedc7bb1a99462b623e811b9ae8cbafb2663e411063bd1a5e9368", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "b9c41421c3b21055e821888b2d62291ae29e1bb14c3a27b8347053b96ad82c4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "90446f10d2330eb9b5ef6f239f5c12b02afc828bc028fe6ba443e6d34bc9fae9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "d465adb983bc896e688024d7449355bab3e44466e31f48ab61cf8a8fe0e56475", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "baeed0095a5b379586f44c385d6c44c2a73bbc05fa4421f09ddde29c4ab3c88e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "5005e257737ca6d19f8554d1660e3b229cc4e665cb44bc25d84d5016abb7513b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "37f83c94ffd41de5923b6b867e4464f7d18f5ea4d7d4d010dfe229888fd1f02c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "885358f7387208c467eac006173ac492c99ebd4e3b2154a3117c3e16f5e1590a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "309be1679bfb7cdcc779dbcf3a1776557798b74183c84d74d60e369144b28a7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "ef346f35047446529bafdf74fd7e246111960bc01a1d67eccdbe229a9fc9cbe2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "83564f4931b6bcf8fc322fd80be190c23e6196b3f5d8b74d4b8e4aaaa9137732", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "fc7de09ef2f736e2c19c3e21dc973f85808319d0e2b2b2de5d3c67407e95d8de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "80c9648a717443a98eca84664e67d941561dbdf0b5413a29064781a01d7ce607", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "e4995283972b18cd6e607cbd6ae5187de8903ab88e840c9463532645b78fb82e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 640, "21753d89103baa600b3940612199425f49753e936d1492cdce2019bcf7b06288", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "e714632bf49a176bd26a1f7d78ea6680cc55e01294b64679b813c4c4f9bd61e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "04170436c5b265c1eb922efd74a614a149cbeb578e81665a20148f64ef6c7bf2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "defaee093a40c2df5671f812a263b379e176f0864cb217e51158da7811705295", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "fd6ddfb26ce3fcb2973e13e8e7f93021e0d963f88d3508ce263d556699b1c7bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "b343a93f77069c71968a55f7309324ea644cf0fcafbf3cab63877334767786a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "296e910773f84cc1cbd51bc7c711fe6b318fc3ac3194c788e5cfef1d944fbfd6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_dynB_sm107a", 512, "b3b3a7dec8dc9de3e2cc6b78bbb5a85dbdd4b1466db4b551d4a0398e9d753bc5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d42902d706d045d04e9376145d6fb132869e19b4a9e6c853793235c7805221c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "8ce11ed65c30a414e9342dc6ae89cea59f5e98037555cf665b2c64781a551f93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "def8cd18eb9731ba5524c1472dae75b9a3b66f078480296cef8985b5f728b95f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "f3f548bae1128a284771a6a39944786016947b6aaf6c006b4e548c3ff40ff4de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ed90a89755fd79a53a9bf184f100d2d34f87e9c12dfb2a9291ba4a894d94098b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "86ce2c0dfa14bc0c4246f2273b6f591764fb8478656f96a96ad4cbf1f9ad21e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "79fff6fbcd341bb4b2cec64a1d82608a8e298518266e54d88843d3ce32e2c3f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "22fe0d048dbc757a0a41d547b0234c07d251139ccf18e3f9f3736eaf5ef05aef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "6878c089ee140c0b95304f24fe8bf83cba4f6bd49b2beff9e9539e101b96aeef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "48d4af8c1389afaf71ed799e3584e84e5161375343fee92d7af5ebd1a0a5903c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "744dc95282c9f54b00fd140e4f5951132f6d0e5a1141c8e2f8458827d9b5aeb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "dca395e48e2b8f45bec92c4f8da241ebd0fdccc2e761cf4d0e7f1d05ada45971", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "635185fd65afd38271f13acd81e1789e0260789b0a70f17d322bba387715745d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "6dd2aaeb3a8edf04ae48ae500b87b00d0ea2dcfb529b827a2e39c4f90b05eaea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "6f1322a342c6b1c9b679a6c82491e18b631713a94e5cf081a863894d7ca7e824", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "bf040bf53291ebb8d4f83cfee3df36bd665b78d550319e68165e03674b4ea9b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "8ea6ba40736323d5ad15ad083ef10fe0d41fd95ebd8c14fea98c1d3d3d47e49b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "cad6ac7f26c0d7bc8314e2a2b3b77eccb99aa4b6338f0749cf433019476760d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "df2f3b8066a40d885df846574cbdbf907a9759c5f92e9750fa2da40caa941e75", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e74b18501b04eacc401a450af101374fd07d4a17bc1160b0ed7111a5463bff04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "cc0e5e8f2672f329810e2d667d1476f4d50e6d94cdc3ebd34d28b79f6ffc621b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "f21b91b1b25de4406267f8d7cf3db4f914b6c4d8bd62eed39c8a3f037e2f10c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "9e27ee4caeeaba019118e6ec44b4e2229f4172d6939a3435548e0122f4362326", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a2fed6128fd75f98afc7bc6a74e6f677843cf69a151b374159c263873c3fb048", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "5c02607e4cc3d3bd8495b919ba790f0d28b15c4a7eeccc1e6718c1f083f8ee0a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "3d6671de79b0c4aef00aacd39de8089980f5e3febe8dc6d330a15b7ddcd60917", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "d1d7be9c10194513ce9d796b8d3ed5ac6258a212a290316efd6bcf32f0052713", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "f5e0d4741deab725232300959967a308834c89341c550a9fe75baa9b4ffb9399", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "093624ebdd1ce39470c662305a1e9d6420993af104f94f9ba17a1691bdc18047", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "c5fea5240813a7eb03231bc1d409407806f6b097039e76034566e80145d38430", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "7c84a1c2dc875212ff0921279b17b9e399e08d3f8116483f150cf44ef15d7ecd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "be408605e8d8e7adc989df1422b44ba1843ef2ed666bb46e9ce177c812ed977c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "13ecee2481caf3f045f0891d7a56397c73e8525fe6a2b1b3150a4761fd3480ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "a3e901ae8467fdaf4928d77ab374c427b926d05be9a5830299176820c28b4230", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "61066809b52ed768f0a843bfdb403e028a08ac93fe4ff72c850c625b56599768", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "5ad8bab872bc32bacb08fa996c42921541b94cc8cdee78792c18f2cc3289119d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "89d9851946319b8b6a143dc827d278ac5e841e714df60b0e84205ec6746aa077", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "5127868baabca769a723fbd61714db3b3773543030be60be8abc85572f9fa632", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "dd8345ba60e44f7f881ddd0a53fa56be5b60563a786354c258139d05a82ac89e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c3284b77bfcdb26db036739347e1718afb6723886132304e7cb72056a97926e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c2d5ccd01c8175e56e95f3269eb56eb63f65486b2c084fdf00cbaa248c3fd611", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "518871dce09e532dd036675164643844c9a97e6b93395b163de778dfa71b64bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "0b2164f92716d3f1ed6c42f78c67956710f48f4a6b2b575707b0e7976c7f59a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d9b3f246018abd7ecbf8f92d3603ca99af3e79214f382c389269551e877ed175", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "811d2be0ecc27691e0c844584a6ee92bb01f3193859119b8fdb599f7118ad66a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "01f7b12578a95310eeaa922748960c28405abc36b61edf81a5aa2be62ac40850", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "af5a7364343870137e4c659bb1dc86800500605b772df73003b061c775bcb623", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "30b30ab290f77e18229c29b02be04405d0f1cced863707a5be539080a75285fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221840, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "73d0693354b9edfa11a79cd93648ebf08dd88910e7fb1d0a8838268312224798", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150032, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "34bec590e0b1f21f7d2035b9173fec024950593901f91fa9267956eaf9c31f75", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ca8d3bcc9714494950b192d19a45e4ac51c725664e59e999e3a0e299f4ae56b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "f267497cf245afe6e2dfd5939ffc4319a7ebffe382c705e8ab6d62140f53323b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 185872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "f1e653b677df6a5e931857b3c95f88fab1c85dab658d4a93a0191d759205029c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 150160, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "38f50ef3d0854841504d829c6b9d5de4e958854f8de59cf4ed85c02e83a00d71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "973e8db2c3d99c255f98288b92015ad075271e97ca70f43bb4e23982ad4697e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e4b142fd0654071404dcf3503f5cec8817a64d6b928c9604941f9ff0aa0e5932", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121360, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "dda2ca083d2635bb5d7c7aac022753b09bd271525a69b1f9c389e5249f9d4ecc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "209d2bec6d99393bcb4d4c539945871912fb296c01bff68f55b5c0316bf16a02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159248, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "05009e41d56a385536aad9b7cb46afdcb64666f178ac2d302048798fe624f68e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121488, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "7dbad1db0b6bf91d5b7143f1b8defe8914bc8ba465451acbc598ac2da255b7f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197136, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "fe2454365f00417d4a6484a7d1ae3e379ae9ccaa3e13d531439e3a65c621df25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159376, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "7bab2ffef01e6868e8f1a45966fafa44e8eba0bcd4250bee84194db8da06a661", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115224, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "70b76d199c8f211b22c1400b288154fe626c508792fd3f2602977f42e3e212b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197520, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "eee246be3ac189dd141bbebb891486b653bf5ec90057d83f8fb6f4755c7b3ff1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "35c745965f254a0103098c6e3fb1e6abbd3510bac1a89021ef3d6df42885e284", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "9551231c95c569b3c68241bcafa496598a647a68d6044a11c758774a6a5b3371", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186904, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b91dada25f7a2b40df579bdf4afdddab80d98fb39d82cd04d66223b2cb8c7a12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "3e08be0bbbd4f0c779e13c1d1bb4fbba6870c7092a93145877e4903373c6c098", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "4313b2f2e6be3df7a9214ef4020edc93e886600547545984e16a64148041f151", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223000, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c1055ce6165fa25cdd6ab6c71a325f923e9df652686c14447ee9e79a708d6ca1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "aa8274e9f96ebc22db7382171af7fedf44fe0ce5e5258bdac57aa2e822ec0352", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "5ac070afc343fd0e23893d002669e01593d79be52889f5d2dcedd88e04652047", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "efd08acf68b5f4e9ac19d5e94f319c740b0996cefca76770c81031f815e9d51b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "0a0eabf29b71cbcb4d03708db87672448d025d50c1011327c56c159692f12234", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "85e5020b966f3ab9f626732d708bd562cca64b35d036ebc3d6208ff29b4574d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 135824, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "475271d1142205706c05db86dab2fade6dbd17badc8bb6cf77ead34e8719394a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "99dae5f31519cf56dcb4bba5f0cfa5552a7309c93c5e3d2fc91d9a584318b187", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 221072, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "c1b8a7b97e181df47ba22a1ecc3ff72350c0fe512264968e863a643efdbb12dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161304, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "14f594fc3ac78d2e83cdd197b574716795848acfd76451b1b2ed0d78d11c2040", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "011c9a4c91544138afa14af709cbdf3fdbd0296659d6ffbd43d7f0c653ab8092", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b22364d3c66965cbb037ca8ca1b02d913f340f84070c929a0910c6c1ee7d8df5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200216, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "c612f63dbe27ef0d6ff93452e621edb7ffd26aae1d0735f3084dbcbd547001bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "2ad8a6a7c14ae86c517b19739fc129ff712ed8a545bc9925b7e135127fea90f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "770ccc34633cb759469812d3f1ecc3e054fe1a3b0e2117a1825be7e2ed4212f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "cb7092a09f6f1d84b7b24abb760930a1efcfc25ac96cd3220ac7f2472c99eead", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "79321132c6f29899e0f9c5b5e34de47452e5932db3c537989fc44d75d16aabbb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "15c67ac7e637092a55565b1e31c206a8a5e7d5305828860d8577fe92db8eca08", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d1cfb9c10c3bf2d4656575fd18e363fda95e5bd3a79699e34b8de42b119541d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "f0aa08c244bd6097f487555920708eab48fdc9d276961fb16f197279ba8956c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "df9bfc95fe1939c4d635ccc82e13abc453d36a6e1f9156fa148573c04954e09e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ca2fe28f110b8fa59401e7ed1d3bbffb6586e40acf8331de6e2bc2b1579ee855", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "1823fa50d394b579e654f5ba94078233f8cb4d5c92996e522054a4c9932f30f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4eebc14d6aadade57569cf3cf4cc70ed15194c3b87878e83b60ba6b3bfe98a47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "10d11282872e3ae8f27d8e553b7e668c97a440f3c9e0616754365be9e5b180b6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a8284641ea6f94836d1aac7b1205bc0f7e4a189eb9ea794a9c7e3c327797de1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3d198acf54ab7832906bde2376dc7c342e5c7d23e1820da9f015c2d419b7fd59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d4fc70abe5dd8f2a58065df39f5ed7388f701d0c453bf7cd6236e51caa913a70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8cb2b9e37d31b1e86c23b3a590f8452e8e1c43a82dd97686fce07b4fa788458f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2bf60da49e265b8600e18c9566979fb5c26aa4c5c251fe1c06c49454a9844297", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "03f1ddde4d8db83b8ccdd69fb65a5f04ad51a4459f2468b5e711b554b9432d1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8822a7c4139bd82258efc3fed5ebf17145a32b8ae46917e090c2e66005f43e97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2461904f236d80dea42e6df48e95e3145195f867f840613d708a0c51f89a10af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "33c47a777c84f442f1c62764f41770bfe0722107604f7ab0326d312c21f945ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "c132bfe0d1255566df533b3051965c35fb1e239b94a883b8f8cd50c2079b41a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "fbc2fa6104f10c0be870f4974585fbfcb4ac2386118963074a6e8a5347d81b38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "cd81d569ca8322c5d6de383d2ff0aa11104a84fba441fe76d43a39f40ef30e01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "77c1549e339d73ad6153eacf10d46a183db25e875088c93402f5fcb9f99cad12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "60584f5727d37cd234d18965f15593873f45918fdc338f018fdd8516d3c1094a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "63170676849e929967fcd4d3fd605b6ee0cb78858111fcdaaf576b0a5ecb570f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "97598560565b9518c0b33803f68723853305cfd71c9aaaace9dc3cd3e4e53e01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e13dd40ac89d920aa5896f58ebc41f57f108f757f4b9b850f6b844501eca36a9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "09e9305c98385810961467453f80b1a34467b79a8507a4201d5ae9795665e2ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "303cfa1894f0a6e3b94fcf797f318582f7cbb9ab21d4d82a437c0303d7127d30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "490d2fd88392ae39b56375c04dcbff56cc0cb89bd438b4bb0c114c25cef33152", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f707d79d0b95365e3c49b020cb53650440dbe1fa6b91ca6514594d3c585a602f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "4030860de999ffbcd5092fec3d13c5038587488f49193e829d98ffbe095d22c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "482736c9625e0c74e4326812cabf2c8f6d1142f611b26ff214da1a983b37397d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e19709a06a7c406f2ece5bb7516130e1a32c610d035404b1607f0697f8cd9a78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "907cce6eb18f1334f316c027d3756fc2d0d40668ef9a5a35c63f4eeeef1a32c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "a9b90caa38b200fc014409a6797b0e9ea61b2b8a5caa28aeec24dc851ac71cbc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0184ebf452e3ff0fe90cb3a506d965d0deee960a79f5b4bd8a8f0507ee77efb0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b937e1758ca01608712aa52c861e63e51fc63850005218377c3d50d3213d3967", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "216f41a2cdd3f0d1eb7d9aa988760acdd248d357d7409e39a842e53a38aabec0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "64ade4972da3f1d76b26d1ea3ece674db570dc3c4207d84a868b1149c8a00504", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "7f1c4941a57903b12d7722579b5449bd7d062cb04a03387e4654367a7006800c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "f96f9c83e143dc69831e2d2bf9c70417789a987779f46d5d14bfd59c2742d6f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a395bff737279c7e7aa37f34e69e5db9a8fdd613e905f9e951878645e6dea3d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "206a2a212b7b830380bf232ada729e31434311612fdb9091e631b039e9662dcb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "5f328c512e29d61c677f7ecbd68a1d851fb90c1914249b3acc1de25786b4b937", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1eab785f6912d0b1b26daa856243319c905e0726d624383b784b22e1a970d5f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "43950424f9616f67c934940d7d5804935443d8a47459a195dead582707796ae9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "05d7872d6989a3412a4caabb7cbab2c298d8bd49a83582c6d093e372a6980f33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0e90ea3d02412906b990d49df55269ef8ab8bc402820d120cd65534e526a90e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "03ed02f7e57000e037e3f0fdeb9fe39e8b33e34166f9fe2066bbd1b3b7707f2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7973422e8ba80fcd599d34f772bda9f23baedfe3aa5aacd7b63c05e71376a426", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9baf15d588e40903491b6596a8c67fe719fabe39262dd1598eaa77f7cf97b77c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "51029f4d6a647b4b4521f9ab313a27864bfe1322946f569f37554e11356ec9df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "abfa0fae1b1beb8fd2e21d745250476205302a516da68b508b7cec2e42937361", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "066b2a372c81baa28db287f515a7d3d5393692640c2253c908e7b6864002c7c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2afd42326cec416fd9177ba783c6a920fb64bb665a8b81f652e5b6e97e09fd52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "fc15dfdbb82c91423be15f4f8394e7564c02944eae4c04e54398e5ea13a04040", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c1783af673bc746de9efb56c8db61207e0f048036fe1ca3f6bb04da3acdcad60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0d1f9b3869410aa4ecf8d1bf182ce7ca008cc11bf4310682b86a788cb48414a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "517cca665db5a5052e244ca561668412ecfb588e3cc36cb29ac534d0006632af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "514f1448380d88aba8d46efdc50a7e120009197a4dc94cb8972a7eb9d5a48ab0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f3861e8edf9552b0d2c078a8a89ee2a98d1dbe3d8e5fd6405fa8f6767cb563ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "f1a04b4312480c415e597d02cc6f238a9bd391b555515254f197888cdace2731", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "38bbe603a95ae9f03328ee5da548e498b314861f879095d1b2e6887807be90ed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "e1af21fc982c6488fa49211af00731f9398a87c1f0292f68d592eecb386699b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "ca920b249c1493cbf297c996a2821c4a35222871463d4a35f4355e273cfaabcc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "31826185f1b2fd39d37c4d75b9ea667ac7dc2be60c7eccba960df9b43d4ab2c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0450ead089cd6c946e469e549de117baa371c211605f612fc4677a62d56af982", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "61087a62b6c3b5550e978acbeb15f4abb7cd4e9b07d716aad79f2be7507e945d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2ce982fafe3ec01483d200ba2d39ae0fef15f147d2f474de69021ade84b4a610", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6edff0a4f412738f9ee1d2384621d04766de182237b88b1ae6d630c7104f83f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bdb3a43912775c98d79c1eba6478072d02df5769a8e3c95c57d25602f225fe83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4cdcf0c2f79dc7fcc43408329f53cce201cb9f2fb42356fe0f5672b85b49da90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6aa7ef887d179d1a26c18b3efaf164fd83eb6b4380df66913e9d2974f271b9c2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "fb4c9587cb5875232c4da23a8edb8e384d46859fb51bea3aa8ed2c6087652892", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "e99c1da322587add0ac66d26acd41ff7bb413c7021176a9a61531c81833fa8a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "2994f6ce744e23b72660afcf429cf19ac2cc5b95ccb194a5a58a878e764dd6c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "94605a7259814677f86af3815320fc0578d03c9c15a10f44b540a9459ee37c9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "43d01859dee78d8b693abf9709331860ab642c58c764e4210bc4959988fbd9db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "8af254c4b456a5ee596f59061cbbd948b2e1539f5d5f6bdcb77d04ae6ed880c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "1ea85fd69857591f778684f3909c9c2c322bc087bb21fd9f36ce12393b381d58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c54d834d5ad5ae4391d8f48769f61a9c07d3784fa0413f7f43382f96c8824e43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "299dac66c0879ceac5992fe9dc62526f3daf4f6f509c0f4e19047f430d447814", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "e8d7e4aafbe7b7190b93d43dea4e23090b672e24d0d94d903fccd59a17ef3d60", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "066ee6af5cc0d7a777e7e9de2f3b63b4c3dcd0b283619248ea3d6538f886b94c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1888d5d595ca4f2cc82fa3d639c81570d7cc3dafc8ec70d5d7216e59086b792c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "a210b31d2d574f2a98fff87026ef26f524e56fb53a15816d56b874dd72ee239e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3523cda486c806fd803fca0109add548387941a1dc374e3351be3a730eb27fdd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "a07c86d342c72328090042832bcec36d8bd670ffb21d2012b9730f37b73704e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "d316c201f9d39754df1ca94171f8147046d9f7621692f1c65769dc208ae5414b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e6323fb52173f0acd3e9c4194f4969bf7d39e797b7226d8fa4348dc978de3033", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9a9ddf53c5fa0cde1813d750ff4d706d102ca3cf8a8c8435944f035b9c1c95a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "b84c1e44e0a9fd56869f7f66e6ba6ab04cd81fa0a4955a9749a6f5ff56e5cbae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "563b138f66fc1afb5e528e91bf47cf4c53e89942883746ab7d708cf9b91fad5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "e2ae60772abc0cdae5abb234d1878e265f7fb51a39709693f44b800bced19124", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "2ee9e28f4592846aa08c509a2a588cd19037a10358ed14615638c9ced8aca402", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "8e62b5d61fe84821b41c69b6295c192c20db25817197e41566d3447fce9591df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "e61b748c381c64a53478a92c9f34e203b74f2276c1d2888c0d246492716b6ec2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "a92ee1797b56d6840e043103d667e10a99d11d34f8d6a282965e0a9c70ef0c96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "4c47b25c9e8da4dc03199e443ac1cbd8aca6e7490535d6716949b7ba38151536", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "45730ef63119a340387a8ce0c1b9399f0e0b6ce2699b03b5cf6bdf8460e93baf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "7836b364df07c80740c1fecfcb6582fa5f38433382b2f43585461fd5269799ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "26776b189e5280e93c933bd9e56bab5833991ac76e60e37099966e3927937009", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "df7f994ef7e20d0b3649f20ef00e93f3ac234af1f5833dceb42716f83d6d882b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "d50e50e32e6b8be1dfc9ef59abeab7221af7063973c5efd5129e909fac382f70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "412505253913cb89c143b4f3b681e3f809787eae80edd8239fdf23df469f642d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "5f5af955c55415be0db1c05006386b36d83f8c816ccbf1ab089bba5e676fa12d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "586cb070f3a51fcceffe40b81ad15c995a91f597b26ecfd23e8c066f7277fbb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "15eea3f9642f612f56fb31a2d210a0a69fb484d815e39d3d19a6bac3481cd009", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "59b594c8ebb44e555a4e9eb64600edc34ded36527bdc20359c1b395d74d71fee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "b73a7606a6800f1ce49a83e913e3e70d6a6ad825d2e26d2b6bb6e2410c8378b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "54e117f1fa515657b93644fd6abce520582c20f8d5a225d5bfc43e3d3d0885ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "1844bce0bf1555df7906c7f43a6ec93bbfc88405fcf8506d4309b49f5692692f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "923d3795454c62d131681ceecc59796b07d75039f952af63aed00e0fdc11d8bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "7e03e4138fb2ddd9a0942b6a5355a63e3a7e8b230957cdb9d99068ee3427c6ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "8996c826f785d793fde50cd0bdb0e0aba9d76872c0a50de561b2686a02385249", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "bae358725af7daba0f9df72486392244e5752601eefd01a3b77a5cd63f03b8c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "b5b645cd752f8e40710cf54563144fbc19257fe41bfc0332b31b537df19c332f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "4b11c425678c1b911e043089b8bfaf5ebba1c3ac573927699c723544e8f9e8aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "b6c338d1d0f97f6bc6eb8bca94a033a54887adbcf388649b237be0587cd3f0f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "df50daa915be7f3a86a4ead3fe2a6cc8b1bb804134411ce4eb49310b94c24de0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "1a5462ab713b5dbf42808d92b518398f0b8f9b59966290dc2c862b733bf15f89", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "15fe18833d22fe521e0cab3ece2d09a7ac04f98488314a3cd141b53a24fd35e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "cd1d1b039d92fb8004bb5fdabc8b6cc5128b329bd1f57fe7ed389d0ba5282964", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "2dccd050ef4f5f3091402bc3cc096d0e4cdcdfa816195f3fc9ae7fffd99d3b9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "692ab517442192f009d18ebcda96e3925e1bc53eaead1d3bf475f8344aad08ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "5cddd95615a085bb6636e2c0208eae7bfdf5b48e1024bfa2131ef998d2f99d74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "e7bb966eb4161b07a9b10d8a0f3af866b7ffd93b431cc6b2a9a10a944345184f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "7c13722b922c2cc29236635d6369f4de6c319075eae4dd38c36d7cf59a1ec074", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "75c6401aa0ac3aa0c803bf58cee85de439f3c152f130cb9dda3f99524d90d484", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "2ca956ea8632d8f3a90ca1adf4fd3cb69b99202f0331da260ca7a51480bc64c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "a1a8c80e5217d6ed97087b4be2f87b9c55aeb7388076dc745b009f171d2df0a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "1d66f5c2567d634cd1f74a63ad82a48b3dc20ebf65871b3da18c658fa68b1e0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "a9a1ca29bd0821c988d1b1d58bd3bbf043f0bb3641dbeba0485b29973ceb11bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "4f2c4b9da764437b09f7b8a4ed97cedbeaeac56a5de7183bbe2d8048e91e158d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "6922c18b1ecb0ceae5e2193fcdb9d7c1a05a1de62ed8c3b9240e413eba336a58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "87688afaed4c0abdc954235bfb6f08042281d8a8ff0206e69d4bcd9ef1e38a83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "122fc376243a43a37fb63e5d8739a11bdb95fda84eb1bfcaf44533761ce2d52e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "a25c2abfff605ca99546a09e3cb32dfbb6d94ce9f70463387e4fff6820e533cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "5c72e70bc7753a5f320140fff8ea30a713d6ffaba2d37b5bc95d51f88b1c87a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "8def6c83b226ca8a31c3265c4e41e976311113a60f0c0d2effdd7b27603e1fe2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "833c4cce3f0872ff5664dac330f81ef88610cdc218362ee5a62d0b436252bd0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "13a8f0fe3d0d42c77aa07cefdff486f0fb66e44d533550c8ab17bd8895de16bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "85fcb374b16b4c62260f2b09b95ed484c84c65affd84c9adc8da0fdc7ff53843", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "2d43507c5dccaf0291b18dc67ae3f288c2aeb5e7f7ee0b361af0b74735892bd2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "17a0da6cea44a89514ec0408150d8ed19a8f76e6f6df6338fd252afe6bc08a6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "762fded99500c01a516e21f5b63461024ccd130eedc00328207cee440f841b41", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "12e14e3d5344d8556faf639a4915491868938baf6af8557ddc83569b47290a83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "8103190f00442a889d0bcf67dd0f3da87c56384dd35f74ba49c54c8cc69f49cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "bf8ee4959321a718e326add89e25e1d63386bea5ad017b4f0f024d032d113adc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "31d88803bdde431adc29796f8728791d9bb5c58264241e40225a4e0a22dd8295", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "b0d76323d8f9b2942109c357791f8ffa295dd380cba53230fe2fdb0fec44e0a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "fa82e62a32b4bc95716612ee11e6fcb8c0616bbe723b94c710febd7f900f2118", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "9a520027f70a7e88e302073e34def7b1c16d3fe64d56fc3bf1bd44a343f77317", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "129628bf40f0cd7aa9834d4bdce58940051a49ae47bab63256fb336e29a4ee48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "b4b38611773d44b766898b044c372456e05cad093e459d87d956379ee159d869", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "db5bd9fcf2d6feb894ad5f610f74feb4268de32cd51dd89aa9f95e4fcfef8d45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "f13df1a90288fb7da58b707d938acd383858523f18d506a91cb03c31d78f8930", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "41fde213b8e60877864efd590efd2d52a81f58a83b9fdad8d1136bfaba8be664", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "db08e8808eca5f6f04eaf88cd8519ea57f1014672346c54ffac37432341a5b32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "9eb379f43a291e5c9c4f337baefdc40039b9d9f0075be7c638aee60b0e23dfec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "1be3fad9104b4783a75011fcc9d51371a5d78717bcaab65b2bd4d51de4e9f0ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "56cde577b5870f565ff7530fb837a49069ca8fffda67508beba539e0748640e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "f19723ddcb251539a79d2394cd9126158fbe267cc1e5a73431337482fd9cd2cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "aaf8d3bb064bd89bfb3f3b1f23d461698427a7a79875995441d4e8455f9908b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "4decf2a7fe2213974fb56d8dfbea36cbe4564b45a15d90a8781d83b413fdf01c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "d0b55ec5eb8c2ed197d01e381ea27fc60f2d1870fd3b64921896cb7369108313", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "7c6264b9edf89e724acbd42a10c50add56d931cd0a70906ffa12895bfa077190", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "7e368984b98408cbb03a99bc3bdc42cd81288b02a774c5019da2f2fc0a9c776d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "b1e1d9cd53d1b9bf9780b61db4fcb95e3b8235fc0f7b20db30da0d2ccc153091", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "a41efdcdd0f8d0ac5a2f1dff79e6a62b1d06cddf63c32a8d9183c016431591fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "6f898e31c929923ae47e75a9b3660eec472b92c2bd812868a31dcb8c177c072e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "aa7b553b7174662b403b465d13d1344c8bc709ca20680d2f5089da2035c6d654", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "4c59f77db0b8f33818375e66305b2608425a733aa36c46cce5539f167625c0f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e2bcd0e87b2c70de3d320d29f3f3d7be653aeb3129e304fc4301d8f6973ecaa2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "409c6d28494e1c31057d89efcd656c763b8176689a7bd13bae9fb2d0f9d1bdf4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7880d6145b9887c60d76142b788822a73092e227ff5a4d3bf6ff0d36eb186a8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b6696843ce8fd838063e4011feac69eb85d724f9e5259ff95d3b1a600d026a61", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "bbc5ff94e0eced75b664cd6e4a2ace28a0bb4392d30b04afd85efd2a5353f368", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c98e5c00df587b91a219dc0a6f882d43b4f74e274e1a0e74cdf94ec5cb0d4762", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "9c770723447a8b74d3d2eddb801aa177277ea4fcba89c672aa200bd165706ebb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a387e07db851b1ea69d1e40540482c57ddf324f3755c6e2a9470f488d4d7b80b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "9370217ca3c1f9ccab1c6e85e3b77281879aa9dafb74f540b282be3d747fc51a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "565692f5dfa099cbeffd46df71750981af3ace9ed3b64b7d6bae56508ffa55b8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "3808c663e9037ba3bd59c134f5b02ada48c61bec71325bc9db879bced36e4916", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "19323b657b0d1db79a5ef2e8ae807bebbf0e432053799b8bb4ac42ddc47b8648", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "db8462694df9275f62a20c378510a75c837b8fddc3ed6424a174b2ad99a1927d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "adca2ecc49e6966f7fbf524dfdde375938bc47f4605ef3f30059fe5dbff97ddf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "458227a5bcd4e1dbf0f8e8769cb59b502f27f8cded37d5273d63c0ce01131d59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "6044d03d86eea7de492f87c74103b4680e6f1ccbc526ad5e23987b535ff1bd8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "7d74a316f39a416c49678f485f8005abd71a38f4f3281a0903da4ea21ff5568b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "ab7ab5fb51f0d73bb6153312c91c16ebdd0b3716404c6032ae08fdeace2ec58f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "bc99471c0d921cfcfd12e5bc411ed01e41177a2263639e0baf06a11dae6a9df6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "2712c462a0e3d6e30b69129c3b2fc84ec4ea1fdea1767b0888106a82759d1e4f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "654ed69ae82cd7fb6dbad7391a47135a19cc16b51774a0b75519c493de800945", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c877d6685f0b0c6e132f4dfcb672eeb58c8aa4b8a6eb279f485e332876459a56", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "cbfd460e83f18230c6fa1bbb6d277ce1d2f7e435210026dcf7f28b9461b2974a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "2b71bba6859b31f74977ace34edd9f790ba666253ae31d07a4592bbf5bb13ee3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7d5dba07b9d9c8807e49c485b76ba204090fc573b8ee8aeebe7a3c0e4d863328", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "9340736ae4cc560f8f436e50c2b2132a5dba9233aa6f6050f6a0358606fb439c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a975fb5388a79cd0276658dddef9f4b10a0d85decb31dad63c09e2251c793b4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "7896f5d4e25da0e7b9c1e8b6b59ac3ecc5a97072c6ff3249b7ce7f2b3cbb2e2e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "7be3e4e1318b63c3510903fa0b8e05f3dc04cf6307071f0eece9576c87e7f138", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "1fa159a214ce023c7c97e8cfc2d27354d824ec3ecbc25559b18909b5b7f90426", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "50f7a3263fcbc77f180a3cfd8be1f075c7031360a84a8f729c2b600270e1e3d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "196974fdbcc012bbec59465430a534e6d424bbcb15033045ccd522ed7ef6943c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "43295004b7d82ffacd47be79b0b8be5f524ca2e3059ca2590f98d55adede922f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "261f0a850e28ae4d65e61d830ef32697dc7e9b9651731a780ca92c815f30cd05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "bac4beaba33448569bfa1a06e4d175a6e81938ca37dd3da799de6bfc974dbaf9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "fd5e9bf7f3501d49693868a42c7c1c5b95eb90de087654a776c68923e7119e59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "742a8e4122b6ca6512fed622fc07f491654c7175a361c62672c5e602f87a9bf0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "dc9a9ef145b491c41b84cdfd120a3b4f1c932df3cc429f0a409c3a20b594baee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "a7057f677344bd50d06e2dd2802c54c64e14d026e748e94c9e774caa6ba1cbc3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "3fa35deb2c21d5567e9a5ec5dc9a963fba5a825f743fed6b0ece1dfdf9ec1bda", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "26426b87e559a4250383c2f08877e48b429f7410d4f4ef1d48b78af20c93b961", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "3484da8e2629998b80d3ddc2b2d97c3e90ed89082d32fe4e81e7fb8a20e0d348", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "bf814c95d7b5f722778262d8f48fb2e5cb0541aadeb73bb754870813e1a906fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "5a787ef6316819314468b812a73e0216ca427b98ff1414e133ccce8ef602c719", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d3018c47bceb24e36826e164f9d5040b2c7ab2238e6ded34a1e96103d17f7370", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "37195856237b9afb0e3ff790cd59b72593a847360b423787e571797ab92cd33b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "fdc499fd198a7df7a8ef124647c581650e1614e91aa1f3c500d0b3dd7f7eb5ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ef863da353ffe18aa1e8aed9cd86f0a0cb113455d5f9649fa134f999fef4946f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "decdde66b135944282779248ba4415fab6fb85cf7a3654816be4c1c0a6e9c13d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "6cd95a27e4f8b5ba1e8764db9a01df45787d084b98da292e0a3b94f57067f69b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "988e5ccc9ee7d40ff9f48a33090a7874bebee1e8fbe4f42b1eb66165845e34e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "da2776a33c8504e9a51b80216c21a6283dc0b4d829eabd8ffff322b742236623", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "60e293cb5eb068b0ad3b17f5b310730a4860f185149b18b292c3cbb4acd0edd2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "4d24089155436ce98ac929747ff0a3747d9630722b8475749a4e46b4207b414f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "41f8e9b743d26d57c26f8fdfcdf313c1ffeb56c59f82b7cab410dd60b2ad250f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "9d052d4ccc52b12388379fca0f2818b71c17ca5d1fd4f07de6fba09601734368", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "d1a26d88fab6c274ce87160b50787eda4c6fc01a4ba485a4de2178da1d58cff2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "b526c3d8ff42e0ffb852cd834875f261124ebc188d5f38f71fb8cdab857d621f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "42ef32baa2c803d3f1a646fa14b186f8782b199d061e869bcc13d98fd8e28829", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "7462228425ac71c36d393eb1f3bdc5f1fbcd763a6277e05195931143dd6644e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "6a7d09746d08063348e39b9f563bf9c00556b76c16150583b753335a3091a89c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "327ab663dfcdb74eb4f7c4c3272b4ef4283ffe1ae24756d71783a7c1f5e7daeb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a9575946e7e954b72e0498746871c5aa443deb9580b8f6071c52cc073c19b1b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "39ba17ec7edd084c819501ac9dd5b38b6d7b30fe80a9a1fcc307301d6a63642c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "139003029c9d011a22a27abd6746bf4e9cd298558991b4ab3a557e3075b61e97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "3ce811219fbd0d81dbb7376421a6c3ddc6e69e986ea17b0ed249da838f7a6520", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "90fc9ef127226df497c8b9ca966de5c8d409ffb7e700131216b15ba9e726b57b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "204f9860ba160757969bf0570be889eea9363479b979eb9bcb70400a9c7c5ef9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "c9873c9a968f165a58b51de4a8846b3b592c80fe12ebbdb6c206f99df2b95c3e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "6d61e744ad94a11217be9f9e500e3ed83fa5c3fe9383d8d69f0f6e8a1ebe3533", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b98ee9f2084669245ca4f7c4f4a97c7e74a7a34b633af9aef1306a70464aeeac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "cef2ec4cb5f8cf01eb34debd793069fb16bd8818dffa0ec8e93b13ca0f547158", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "51f88d6b9582ef22d424a1e43960e990199e4bcbff7dd6baa837bc3278a4e7e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "9770f5347a042fd32b38a1327aeb1d25f41b103af0872209364567508a931068", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "9b942bdbc69d41c2f6400d62260c9a95ef066235cc121cf0a5a374dea5db9418", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "aad02b2667b50f9b0605edfd23460b07a186c8c2e26637eee498235bc6bbb892", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "92556da612b21436ef2ab936c4466b5941d65b90daae1f8b13db3f1b6b16d05a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "fa9d0acd4b2e2c629df8f2e739fc959c3b5c674ecf7d398ccfb14ef70956dea9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "44e6cf6182a6d324c3e2d7606608b1f433a9064f36e1cc4ed0f9b05633f0177a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "7a35fe1d73a05f8eb18006143c4b7e0e31352c3f5214714dcaeac54b1da9ec52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "9f6d5344b8de5d44475c0100450e97dd367299c975c620bcbed6c850352f9198", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ce2be64f52d301a44f94af1579dfe7ce99d80389a5ce0a2001f27e2644e2b326", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "925392e338aabe31a69244d0b9d3a9d24c1a46b13b7b111cf6f73069e6fe36a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "6fe3626d9f3bd60bec5762adba6ae2c4b5e1c7adc7db39deedb865ba64914b14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c92709331953931cdfb5f4af7f811d290253f54336367b1a4a25756ca554af55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "14b33ead03ca2a2349a71713ad3bbb40c192a43f2373bba36c63ab6764efd9b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ea88697504d20ba2c2efdb2a273646ca9d29ff6f97d72fec1bed3abdec9f58de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "876939e98328b84a8f2256d1455a6e9013065e0ebecf400e000fe22dddc0f548", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4e5f92d23be25be5f9ffbab285c6c3233528c5d95834fce6323c534f7d14e6cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4601f54dc9b3516513a608d709e6c9ec82a03451cdf2e84f827db9ce14078911", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e588d493b2751a57e43416eab17b4b96f89b0c9632476798489ae74fa704d5d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9affcf0fc2eadd24ca14cb619768a84f56d992ae0566f57058519a126c256102", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6d0bae8df442509e71716654638b988f481e3322a8b76cb94861f4c4c681513d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b33dcfba4b4d9324182c782fa78a49f0d21b5bd3f421b34c216c1ddf16ed4a36", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ecaf53c92f660bad6e1b19231deda0379ae3d4ad2b87886a62ff9db6c5739cd6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1ed18e998a471b5f76868cc58adcd24339f898902eaf3ac57afe3ac8e41adbbd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3ef722ad8d9f54a9d5380fa336eca5a9379ded267f8aa8636a488727340b3e33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "9e1e995176ef18a999e2885177140853f5c67be2caabd4724f2f3bf2e8ee4844", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "88f399f48cfc1ee37a48072575fc2239c2453845937175856d1e50f304684a87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "5a46203b8bdcd07b3f406ebdd28d8928a3d039536b6c19b1c391f1455738baf7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "8e4a7233952995ffb5203d176319222d6b3e9804d65efc18fadf69ae76d50a06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "856e4a9f5f66a37c1e5c44d680469a32988d1e29f7d95d223b801b444e27fd92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "14cfb442dd6cda6c44d3edff23d518a75a3af067dc51737c68c8c78b0bdafa1c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "be0af522d67ef97070397d88517652435cb9a59616401ca85615b1f2b5c3fa84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "530f43ca683719b4d6ce0cc25db4867631d098ac8dfc7af31966a4f506ad8e7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0154fdea2006d8181180d9c17770750f0a25995940e6922905e5d0b5ac274bbb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "10f6ef6e434cd328225d09005c560fa3a040a8a78c937167b7f34af0468e64ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "38ffd54f841305e5e9910928ded6f7467e7bfffea45cf4669bf393a090263ccc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c420b3059392326d8f0ddbfe0face4fa102c966c8b51cf4af5f9fd2a3a470713", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f802801850b44b74f502776e4dc6bb49911c30af14061d83dfa8d2a253d7ebb3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "6813386409f938044f060ef01c08929c29fdb8058df29874d26c172f8a6672cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2b71f1b21d5e9ce47f0da5c39b591e6991cddea1512564386c45ff7b249916c4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b81bb896f56a07879a260c5b651e261c5a67fa2e009a9bf35088aa8f90362e76", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "a24503c9128a6db6ac140e746a4a19f4eb52858e2758c8e073b16a1e520356db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "a502bcbf09aa84a3996fe6476c4ab141a7cd6217389c27fec628fbad28661caa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "c8cf289baa1889c6643061768fcc8ff3a3fe425c580cef661ca4bca7587d715b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "55578c73bd8a040dc376edf3d5c1b10bfcea91ef2aa14555de03142915497069", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "95fe2053f7eadb7676ae7a4ff554cdd0dfcf716cd73908a095a84c86637885e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "36d24bcae4d79610aa64c1ca1d9b32b741cd489ce8b4783ecdc85630d760e297", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "96a18f7239a5a3277f6d7214018cf525e0a8b299528462808ae72ddaa8d43330", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c1351774278bc46ffd7e086f5d6341efa796514d41e19e3970b75ea4dcce8777", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9c2e1cd2f05aaf06c0bccf5e23b25fcb658416133fa66601f72ca11f12a37ea6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d923a5afa1e078016f5045950db689172a42345fd9631a020262643c4ec9ed87", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "6aa6138b5bfd920427bc7e16bf22c39638e63e5d37039d3b3dde843fbcb64c2d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f1ba2f8392186ab28a80b4b69834492e08a837e3e534c6786cd5d3f2a2c73e47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a85014740f557c939661ef9789371757c58c097304a15ed00c1beffe8fb8252e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b143c4ab79901623cfcb38818bffc8706e561ee83fd8e9116d606d71cdb6ebd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "54ef0f357a839e157284372669d2eb04647f7dd8d3a340e0b11162e42be79c1d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "187dcb678a345a4716dc4339d13a7f434968031379e8d5a6486631995c26005d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9a778793004b62ab1c493c965ebd9163f0bcb8f054b8ce699c78907623ec5d77", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3e7bce3893f03f5b34dbd6527c2ac89d1521e96bc04ed9fe56dbecb16a5294da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2f19d644e981bcfb70d4964946632739ee049b071cc7b6822a1defa062d4e7de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d7e4eb34e0922d4c2f379eb4b65e6ed0bb70f4c71b892f9f292600f3a6e92c45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "02f89f379c7a5d24c97657e21c8165f1557a87b9bd130bbb66479e6a6a173414", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ee1de4057e69c4ca89c1bd41ae7f28b891b7e31d3f7e87655e610b3c508ac386", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6244e72a55ee011ce2f07d17d135a160b74c3b9dcb5b79d0e654261d7abfccdb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "cf1e5cb9bb9945ac2e508f223fdd1224331edfeffd4b1c7de01055c45933a167", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3e1c17ff966913a3f54e5e0caac2efffa84e2e8c24c849c5d2615db1ea3088ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4c0cc86fe81df0eb394d7a131fb5eced0149799b3466983f3b5a083d698340c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "041dc5ff94114890f650b9042890d0110bb499e4fb862e967e81975019caa9dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "37d0aaeed132cd1c47d73ebf69f637fa4bf8d3b27d1f55807b68b1b6032773a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "fba2a96a6818cbd2079a300bb1436604190e46f90e1379b706d9e61f8546542d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "788ce07db4969e27779d84c6b69c97bc2bcdd620adad3ca96a520579ac305abf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "818cb838d24ef850d039a9620794a4971007cab593a15cb559369c1e0e342613", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2cbe52aa57b0b2dacbcfad2f749094940d35762b7b530810bf33e7ce3ec6813b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "5257006e6440896c275ffa4a17cf5319f87f967ac9764e0705f6cd1792d2733b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c0ef71e2e0ae235d61e5b5cad84f2cd144412969012ed3e851b6230cbfc52a8f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8665bb70a22a8c14f0e7947aed87eb63d28df22e952013df62666a950afcdc7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "340a108d6bf1c745388c0e893413d30c65abe618ea0df302334b8e849e99de7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "069556525d70da0717e9efdbe62334c4a2c290d42b1fcee7172e23519ed8c57d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "94ad0bfd257a32741bc92842cd88f4ba1d2458aa9262f28506a46e10325ef2cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "59bf883925d4884c96d378b19b4b88969daf5e9f32d9794995417a781e466dac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "bc694dba2539b786f6417e3bbb8eda17bb5aa363bd6b699f415c4f45d41d2ff3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ef0c78a7a421862cebedc781b8998c7a79f3d29965e0bc772604969e22ba53d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "ad60ed6375798e78b00824d1462d192b3f510e979f836051e857479128445656", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "eaaa8252fe1a2042e63cb82992aa820babe827db79f4c5f3cef4b5debf4ecf67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "f958e8baca6e3dd0bf9a5df6756605adec60efa4dbf63e9e54a7ba3f77ff7708", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "1ddebd2a010f2dedcb377598d0b1a7c7397afe3987748700ee314a1d40ece38a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "06a40a99a8b6e3fa6fc5867bb9b7b0cef51cb995afe0abcd0a545ffe63ced452", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "1232d8c43f370bc0692476ffc20425cf0a6e08320497f381f0b67c0c55a21995", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "0dfcbdbb7a5bce4d506a57d3adaf7e3ef5d781131247b29ac8438f7d944d7c04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "b6b793039f535b60bf3864a9becde96d362444e9d46f97224fcf4d01bdcda92d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "fefd77b18ef1c42b3e78c2916d3d27c6ba158371ce255a43c8be6b3699cc8d79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "916fc49aa191a6f1806953da5463756fd6ff1d9bac56eda6fd045c6d4df2953e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "113b6f95d6a65374863077dafba8d2d5761fe15a97df91ec0bc61919ad008a3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 640, "83f4598576621063c4e75223c4942ce757ad24037ae1a8969845cbbd5be48a4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "18b0af2bd1098a2546e1d6fe69ee17d47c4a2775b31aac79367d4f0b9715c72c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "bc8e59515b2dd7b38ad78844b1f87d3a11ac3d5cfe0ef0c61250ff2b090d8112", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e6a503a2acf977881180c2a9a5a93e230dbeeee117430b37cacfe9e1a8630421", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "16be551dd0544f0a1b4ef371c47da8bbb27420104b1d45cfb44733563913bee4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "7feba3e3960b04eaecf9db70cdc27cb1e1b8ec21c3573e1a2b1919179431952a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "64c0c093138329ce8531bc6bbecd417238e106ccddab840206e2e8bf89e22ce1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "1ac878b79317ad31c297e323523ad74ecaaf47f4b593be820d058cf9ea1707db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "f1baea5debd74f1249d9918ed60aac2b44a0271f71ac680952620e613667bf23", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "0b61b77190b4c766c59e935eb6b8218de68f1ca7bc45b2c3684dfd029cabaa35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "fbe28899f1a2dfd79bb420ddef192e9918059aa65d3be5d54cb3489304818a20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "ed332509b2466cd1db5d3e2dd39597ece9edb373cb328105b7a33af7bf828256", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "ee892f53a710882475a261e50f7bb7b51d7f8131291376a7474b365307c5d552", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "0370c018b6e2c7ae52dfb186d4e2f4c69df6ab10fddb51ee98d816ddbb4113fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "5d89ff9d3ff9fd4da5d405a02a2d378f5dbb84a34e0eeae6d255bbe6d5ec8984", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "333f82a9cd0ffd0cc0820206095f85a7ab851d3ffbf20fa778368be489164382", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "bd26893e66faed0c9fe577428dcb505368d979b8fb29e5346a314203aad91e11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "ad335452ceae71139f946b487873931908583364b9b62e7fa8e1068d04bd0b02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "e6028a2180b76452e397523cc51b5922388bed8037438dc4e762be690de6954d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "f29421088436a02c5b397c5207dd9c4d399fb46d62de92edda72208e08bd2949", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "59614fa43fcbe3374875aa3ff64d3f9cd5a17582b3572404ec215eeaf4f9efa5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "a8f73271c5263b27b2b4ebaae95aa5ae6f2298ef1a7db0b35192737a50ce680c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "04bf85a476e6a2ef78c53daf34d05fc3949879a71bc3e0576b947df92a853cef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "b759a3367c9d526bbc1e25b155973befeaaefe529c3f28feec42c45ec6b847e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "92a72d2cfaa7b13a0479c76821bcc8b628283853a15fef3b8b1c12908af0ff52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "484be99286f6390ac3a04dde22b55f51c18924ac949aad934ad2c3bbee3c0919", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "0008a457ce2f1f38c548438fc6192a08b20eb8219abf5b2177d781dfcdab9455", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "d4bfa8410f6c72d1f9a32fa5f9cb65118f81838a9498080a609fd5a0a902e180", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "c71466d9871236d725515bffce313c6e25aad4efed7d3932d2d4824f736051d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "136b04a068a85d1c70507b41cbf7a3428222ab2d069bb606b1b47453942e4666", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "56e383bb7bc9baf8da5c72fcff8d7593a697ddca56003d7e83c3b37b9ea57cf3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "2a20323b662d34a9b582de641e7b10df3e7beedd5a8338e87f7a071108fab75d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "25da9337a4dfe36d73ebd0606dfcc76c754cc9e8955aea1e4cb7d35941ffae4c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "fce9da5e93f5fd842fc48aae3f4e9eac9d6eb67b207cdd8765b042c938300176", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "e62cde1dc453e38e50ce0126fc9d7e68ed491056612b4d432ed6b0e316d1513e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "282e8dc4a6005eb3c5b03049b73107c34f1d9c38cbe3ceeaed1be456e84de7a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "5202f2691d7a7bb68e20d1f48f345f2cc16a007f99cb1bc9fde354d886115215", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "f3f593ac615c2ee43e10d935f0ebbb200a2cdacb107ff245bc748ee85527fb65", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "512e9d226264c8e08f78f2de3cc62c51b8ff44ddcafa32527c21305bf57c335a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "4424bbf5dc0472144344ff05550ef19946ed4305b63a1fa6dc20328608a9d5a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "dce9dae9c0da4336825508ead7a7d1960c9ea4eaf488ad1b2ae1abbfaca2b5c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "1d66469f500ce6cbb0c97d7cca19d9ca33cd0a0974913bb2b4e93cb454ea1af3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "831992b495b7086da07e303770fca19ade10cdb98ca1ea60d955cddad91ab66f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "f7834830fc741f6f84f93b2b10375c004da75df5aa9e8c74a56061fed1388b19", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "278e5d316ee7e3d5d5bf7f06c7abe2687454830e0a2e96ed56dc776fc76d86fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "8d3faa69d09f774f7437691e198ebcc6526e1cc83fc3e3267ca35d2613d38b7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "aa8a40a91d784377bb2ff4c4aeaa79502b3ae78a10bde0b10716969f54f6db4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "6b513e7c048af908ab67af038d82497db8e2cd3a3494c7f8397ca121b58a0195", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "2e9fafc0828c0fb6c88975049ece58e161b438fdd7117277ce6a77e28ffad75c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "8b183f095900a912007791773d8effc8e3abb4bdc88eb333ab74ab34c86bf1f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "6c631d53313084821ffb47c8439b48ec2f248515cd6db5e2cc43a7a40da8bf8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "f43eab8f405ba15a99a9448f6799ea4172822509dd3e06aabae15e448357c65f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "1165669935d4e28fc10451d29696cacaf85f251587694d3159b47111d2589352", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "1d7a63e0958d6ee55f1537ab96615d73c0698b3f0c24f265384277ece97f8d47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "28f57cbc036f8bd2aa4cd42829891101c36c09a5d60feae40aea80951cc4cccd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "ce8fde5a23ed8a7c2a41c7045e617440cf7f7a641196ad5bc3248aeb258dc56c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "c7daf17ca2277cc4aed867624820d2ef6e2564d18a572a7ecfe4355e474b192b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "c42eb1b1e3de376c2dacbb772a42fa231ae035e2f890d678cd963d44290fbf1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "7886f6a8ab240ac47435c433ea8702fcf80623c4bc90c9cab10302c9458c9342", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "9522ddd93c14be01b77cb43577facd19ed4937198271ad3de82071cb5a31ce95", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "704120339983a1b11bcd9c1b92fa31aa1c9b54f68b841955dacd519124c13a30", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "ab009893063e92db598e159f6f357aff7dc4911a07ec5a0bd42eaa7ddde91492", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "935447ee05cfbb091ef7be004b79e9f00a5ba9042e12880fe490c3f499e65a22", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "d799ff849d29f0969a71ee253fdae0b63bff5181c8c5c4f6d51d5a8b6a2aeb3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "eee62c1fa2b9db8e42b0b3948f15d685994b0bf1db218f5e063dbcbcdf8d3764", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "5df2e826002e02cb3d06f78b8e116c02f4ac090269b3cbbeb00a6b8ce41e8840", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "17f1d0a50b8bf68545413eb177b8c9d2f2785b9efbef4ad58a96fc2a7002760e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "2d1c194e6bfb2c7d33fd06f2d599a4ae1e58effbd08d9fa49420cda7589aa399", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "c15e668e311b6309455e7ea7a566b7d2735cfab3558de2885c872407c95007ab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "f5be2fc798cba4ec41bc543fd3a2a6bebf47211af18213605e4c572070aefdb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "20d89817217a9b0f891eb7c4c52ce958b729233c0dd4a12650be76ec2f329cc0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "07307102f9aaa46f77b1d8c6487abf7a2edb33c95432e0b30afd794e70455dca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "abfef26b87d947af4e33cef74a0baf53ecc94bdac5a2bcbe808925ab785872cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "0aff8cb72be15dbd8c28d83ba6417996695bc306309f7b103ce6718c28238f01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "769190d522a9c2f099c557b32612da5fa7472bf6e3887a681c3a36b94942c7fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "e7517506a5b36d4f85d1ffdecb7cf616f73d7992f9fa0297a6f548bae3735e5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "1dce168918bc9ddb813f872d59cb74b80b043bdbe242dec0d3cdc495d40fe372", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "5cb44a48a7a0e1f73ec55ee161be93f629ca4c1411944c48546bd5991c4fb7c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "ac88e7a6554959a08acb00692d706ce77d086a221ccb4bb7b02422f430fea21a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "1ea213591b4a95141717bf043a8fa31d5cffab7400b36a38e6cb989e5a1fbca3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "f79e25533ca0db02c1fc148da356a6992d585d4c2a04863ebe225ac06bb87766", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "bf97090498601867b83758f7220d952384bd3177d909692afd28b467bc863fcc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "f9ef5daa5a1a55d12e80286b28e983d0579e9ad3e0bc9c21bf84a61fd8804762", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "ac50d0519183a364b0b69954c9db63376cb46bd5706d247a8549c02494b9bcae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 640, "7d2be96867d4152a19bbbf149948069ae083bd7566e758418721eefbd74af883", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "ce708ecfaac030e99f3427dda9149bc212711b75cb1235c33b7559cc774653b2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "b6b274fceac3eddeedc0ed85592071eebcd99bc8794b64b0ebcd1ee425a2d018", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "614dbcf08386070ad9c89231be688b1c7085844aa160c2fa1d2ffb9e6e4f50e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_dynB_sm107a", 512, "1c89816ee96874046835ac929c5c1a1d446c30ceb3f51b18ca9f207b8cfc4193", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7ebe64ab4fe0b163d4a788a7edc236e97b0a8d387097dd137d026dd9fd07300f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "ab8f84feb9704a6934bfc053e254b75197a8fc7311881c2cc51311b97f6fb177", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c26eb262feb32cc040fcf7b5b5c293b376d6886908b9cd961c8280ea48adedf0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512u2_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "508bdc201b8e7e80e9faf351f8b4db34532f6e425124b4c19e4dd5458c7cea00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c65c225d0253caa80654a4b4948a9df4a5cf125f9d6aa368a09bb06838660637", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "18ed73cc76c50ed38070d2833b82eec5b64c19667b79e1d0774c27c5c726f934", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "8e457d2a63c373dd8b4cc2487f17b05b210ef0c88c06b7faa941e4cfe14f609d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d42f8039962e51d7b8819f31950fc0e5987250f24fa0c5c958e7eb5b8eaf204e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "01c256263f27bbc6cfa06d943ee834be76d3a61bb749b7c61f1d370381bc1c88", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "281c2c8a261dc2398cb22e4d5e0bc84d01738f12bf180f3cd97a1e2d3472b1c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "454e3af9351b0e49ce25d5a24208de6a009ed5c34292d8e8e0e8c3876e226695", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "6c4ea6f28297485f7990c6b793eaaf21613333b1878e95f059ec8e5b99aa2afe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "0c8237b40917be0f2c333852d051cbade1c0dd2da08a2a8d95d2bc63b9d80f12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "e7193c8dc51bd3f6550469eb6d8d8b7254288d7cc47f3a269eaa7b5074fa0f06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "1aa318ee31704fd3fde06264bfcd84bc6952141e24ca9981389a2fd7bd05642f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "7d01b2c1223b00a0664eea8f2ba106403a794233c6e81e254fe83017f47e465a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "6d3bf0811bd7805b8d159bed56c91ae0aa30097d4ff5e8ad3c989226a633e010", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "24ba1944e8d3df0bfaf1c3f8f58848252c11d4d2a9648a69d1378a7f0b2759de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "286a9a4dba04352523654b4264f784c8c22fb777516ea3141eef03932fa47e75", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "353bbc705ba43f86d17891a1bfca4d1ae42311bf78c6ef2181f28fcb412e6f20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "5776051d8c9bbd665aefdeee5a4037962c9dd2bd3ba3faf59e83a8f93ca4bc97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "1b154930647be8ae6e4b0bc9fce18f3ec931b9ffb98544ec9332988ae2ae920a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "242cc6b9afa3e372370ec290954f916377d052e6b19d5f79a4939d6e4e70f86b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "f53ff082a161cfc04d760d99a9a77939949f99b439178f1911def6d31ab961b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "1b01b36e4bfefd0efc3404fe7d0d7668608c13f0f8f4deccd457ce116d558266", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "16a68e1136f20d8519774f2df47057a8cde814ac22495f29dcce71c0b09680f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "9dd584ae0100ee8ca58e599ecbd97028bda2be8f2e178abebe8add95272e06b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "4b3185e49dae2267719c4503f469cfe4167cbb42b3c931fe032796c7eaa28d79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "1866ebdd2971a38a53c0cb388476f92548382fe9b1944cbae320a91d35665665", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "085848aa652f573ca815a8eb9adb4777009bc076b7d3e2cda0662212eb2bb7c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "83599fbe19ccb3f49a160497a2257f196aa255c47542d9d2dd2c28d8d7791d7b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "1c42733a6c095dd27102327b7e91f8d6160ac9f112d9a34f30ee46314b9c34da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "c062fba631104f511bd7dda8709d6d2c66a811c670a98291f4a6ad3edbecde65", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "4722000f87e49eb97464041399963e715c7211e6d3a66f575a941ad97d66339a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e0bfa7007d36f864cd1c5b9fcf2ea7b114422e0b800f61d298995f6f5b3f8230", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "34404d1f4f17a123892459449efa86d7dc458c648581b972e931c4e707def233", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "a3291470128a84ef2bf7b818e7fc11b1c939130adb3524f828ee5510e2fdaa0b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "b1ffa28c375318637414cc6fd5b0d3c8f48e9c4419d6e9695fc8543c21a87dc5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225808, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d047e7da72fb2845f0a382ffdef6b3ce17b10aa744f56d11cbafa9852984738d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "8c02ea9c0d3b8ddbf772a6b5e8815c557781976f5d9628226708feeaa6c4e251", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e6267995326487750bf468e569df956dfa0cf99d5f9dfbbf328930e13caa3411", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152856, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "5fef48535a28361dd53e264d5ef310fc0304aeb07b06d8c3dc698075af66e8f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "f2d83c697032096c34ad09eafdca8c99c3d00bfbc083449132a2613673e55b53", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225936, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "9fbc329a965df489c2779a12d9c0b93e428ea3045f0a1f6462911eedcf65786b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118288, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "d413dc4ee4bc4623027e93139744faa63c42ca6d4d5ace3f6f49aa2f56e89ac9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154128, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e285c08b09a3ed06a993e4804bf7a13013d75c9f688178252f5af1f0757bfc7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 118416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7dbcd2037e3aa4e6f13666d014ae19b2a76b1c7fd1781d10010bdec561a93f1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189968, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "3edc3700f76edf06e83d13eac70628471ae5c79471c63845c29557ac53a63093", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x512_s2_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "07f9e5ab3f06026fa53b02553fae1603585f5b8ecb713a1da8dc1d5728a8a5d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "2f3067582cdbfdf38169cf27a5bc1dd10091d6946772d29815d74c7ab7970283", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190352, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "cbeda4ca9ddc0bd59368f7357b39a67dae42880c8af920a0f1dfa45d27be3d7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e08b9fca137a430996b09451ff0761668ba2136796178d40fad5132aeaa034fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e8f3803fda621a3b8541974293e01d3ddb0034b9df4c72e5f1c53c13deab957e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129552, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "3cecbb995eecdfc58d4a3a2647d87c1c60ed79a88362dc1603144aeefd87efd4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167440, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "8931bc620afb5983303b44581e76c28d0bcb27c7b599cf5dc9181bf406e33f2f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 129680, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "82712cbf1e4be5dcf4037eb405ee23a31bdda757ff182e41cd0f5afabacd4a67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205328, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "5c4a2fc1c9f70a8b98661c9622a1ffc7c97867da937f7da937ae268d3563a813", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 167568, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "d9738d219692dee265c3aa323481fa94286a7f8fb008cc45f13fc5ee8ae776bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123416, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "75b24fbc54c196deaf3b44bf56e4a8a2cb5a39909f51a7110ce498edeaeea59f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 205712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "ff48dfc0713e98f1f1c7cad435a1135165441893a5695331e420e0bf7a1e99e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159256, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "5fa75fb5ebeb7092f22f108c006f168dd2ef28f9c267bcff08f310fb4e9d5b34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "dde64b566ab7e843f11f001d9f3ea15a618e767d62074322918a250c1011e538", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195096, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "dd531b27f67b5ef6a61d89e84acb2f962a11328fb5b21edd20ae5bfdf0ba087a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 159384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "95f3f3b32c4d5812787fdda47c91d9c03f12281fb0cd786c6e37510f93482f6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "7c1ddbbe3f4807eb7734e02be55987203b71ce897b18fdfc7ecc8f310bc99a8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 195480, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e6a162504345130632641259b97f8478f337fb76d0298ba7c840bc396c00e900", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152080, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "b40dfbdb92b825b072d02ac0bc299a79f39c02e3eabd87cd632e313bb8f4ed81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 231320, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "e3222d7810eb8bbaeacee8085955f73b30a4bc9937a2301f3d15e6fc00856250", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194064, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "9d6210964a792837235898cd6275da6ba678ade76ef181c8f1180fee05e6eba1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 152208, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "062f8e178071b1497700c41102ac0daac4de1bf7b6f3e06d1aa0d5c308b0eda2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139800, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "76c651fb4f1983e01553398a8fd21cf2edfad505d79f0a38ae0fea1675cbc890", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 194192, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "bf553f75b2e3cd6589cc3e4edfb5e42361b6fe9e9cb2583c9b3da4aea6e4b03e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177688, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "47ec233089d419aa963ab470aa088ed858ad644c4c2aa23908b91aaaa642b946", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "0105ad0e0ae7c13d98bb50ef5134e78e37292bb8b7d30795063d33f241c3db79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139928, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "78b7a4ae35c5892b05962dc4c9cd19ae77df16674355fff96d18f875f2f679c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216600, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "afc7cebead4482487aac66fce693d8ac597b903e47ccd4812d9be739a539a2a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "eb192ad8df7c807a36471e75b3bbed1bd6511cc5d3e9c43094f56b5259c495cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "49dd18d14756873ff1721b977567321b72f34af6b19e527705b63056d634ba59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "7236d7977d0574871806206f8f65679c34b0c1895a9e98b376d386bcecf91449", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "bb70984c4cd84efe9eef633d95852c26f1132415a5f775c7ccc5cdd0a243e6a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 216984, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 640, "65766f53d0d9d530283e143d6fd4cfa69eb44d4e38cf3c1a250fcf9536b52240", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161432, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "a8dc376e73eaede51b7b419db82e422beaaff2f5e0e5c1982ba962ff08b0da67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197528, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_dynB_sm107a", 512, "c0a15b93e18faa02dd91c8174109e7788637a94288f8ac9f66d416cb69c53a84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "80c5c6b2e430dd44c09c76818a0944c5686b84948349795fbd48689c14f972de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "d404a2a10e1f4a069b4ec2fb8e88d92cf8aca0161d8fe5b50e325ed47fc3b0ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "6a15dc84394a1399ecf13cb90689928f7fe8e3abd751d53d3fd02144c08d5d92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "beb4f9ffcf69eaaf249497a774da415b32328b75c563411a6d311e5b8c9b5e48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "df3e77a4d4b4e16385a0b6e8a0af7c489166436d43c5c663a03e806dc8d9369a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "1b74df1742f766dbc3c6b7f41cd60502361323481e33f6f5ec8e5007bb40c872", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256u2_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 896, "499ddba5488279960655d1dfeafdb7311786386cef809fff7a94f5301f653049", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128u2_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 896, "9ed68f4798fb1abda8df22b9748eb1a4d204643c80c1daf493de2955c9035f99", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "54864e4fb69cfea8f20227cfce597c2543a515681beafdaee155144304a59b97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "7dbf76e4b67ffa794a5ab53046157c18579f24616d9d437864975ec92c33a1f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm107a", 512, "f05a1a060d7cdd5d179e6528ed24762642d881f5119f45ba1a9faacb45d5a270", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 896, "136c254b5ca76dcb426c752af5bbf85b70ed256439315e28ff87ca74e70c80b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 896, "12ea30ea372bfe72949a5c6e9536d7b95bbab4dae7e695ba76d29509fbb26571", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm107a", 512, "8df779b154458d9db2c89d1a0294b2f9196739604d60e67843b0f7953c40662f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "965988f3b11e27af7d9eb686c088ea1b37f0c4ab11299ce19f7d98c9a2ec746f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "ee7cd981eec9450530e3be1d7b93e198913b368fa10ad0cc3a271433a562e5de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm107a", 512, "076e1c198830185d05924604baa69557ebc68db8ce2509f6a48ae5d884dcf9e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm107a", 512, "adc42d80b97cecb5ab3791d3c1c5209ebf785b2ed4a3f275ab25b520748fc95b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128u2_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 896, "c7b6f176cd6dce9fc87f0d31286fd837d9b98c682f252cfea4bd5403c4a5e2b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256u2_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 896, "c5868ea3bad6fe35f77bdf8d13bbcbe3149e77d2f680b55f90602d5c33421b29", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "b918d9d35b4cdd1c3333b91cbbad8cd21b00be967f54da0bfcd245c19cacd3d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "68f3a0f84246e2fc804bba7959f75b5fdac85b5baf5ef9652c8b6d7c201042e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW1_lsfbW1_dynB_sm107a", 512, "c76fe9038891335675f742147569a7efee706ac7907cb1a2495216216ee7f550", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW1_lsfbW1_dynB_sm107a", 512, "2fe9200959e0881c894801a4dfcbc6adf5c2d2b53b819cdb58fe666bdba490ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 896, "4cd8794a8afb6c79cc20cdabdd817298b6cdfa41a0c647e75a35a4f5ff25c624", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 896, "03379dea48a2bc3c1cb748b35d02c0d96e3cf9ac3ca4aebbb5be0879884987df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 768, "673d40b6c91bd835ad5c88ffa1487750998bcd9be7d526588eef8c07c80d8187", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW1_lsfbW1_dynB_sm107a", 512, "210b01d6f74b8f30c941dbd570d3bf8202648b42c7321f458d0bebf067567d4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 768, "b1181740c84b29b8d3de83b5db91bf2789bda0c1a5cbd104092627f623823f50", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW1_lsfbW1_dynB_sm107a", 512, "15bc64194ed98a8c0a99281adf52a6061b015436d9d3850fb2cc42c98a271904", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128u2_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 896, "eb3196cc35cee1ebb8499fe78c4660b87bef9a02a41ef88ae2d356e0b9ddfe8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256u2_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 896, "b0df09e8ac7c2d50536d186bbd3e4861ca00cb278c3564ff762d3f1c2adc2406", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 768, "a5fbd2e0299d5310ad23257cadbc874091f302faf281033d9eb05fa72a8d1ce5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 768, "ea0879b7303eb979f7906d4a665f04a0f8fca9859c52b7ce6c3f4784c715dab3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "23ae8e8d38d05140a551ad98e07860455cec574a106997205cc183746a079d12", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "b7d43d5205cc1675951db15ffb8a199f63467c7b633d1ed8c8f5d5b7081d8954", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 896, "2e3aa7296d97f0f35b40488170be9b37b607dba38d547d5f6f79e845c29258a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "092580511e5891b12b52aeff2452757030b8cf47157fef290527f66a28f0ed82", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "8fd47e33eb461599cefbac6d91ff085913cfbdb16e0bd3fec14093f63dbe8e77", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 896, "3e481184e3744378aa386650c2cd272b20b04c47652ef8ffbf607f9656593cec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256u2_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 896, "86c58822f2fce0bdb8099558e4afba437edb39f394fb74b34d22c3377d800390", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128u2_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 896, "d33c1dd4ced4f1cb2de9703c439e9594888276edf72e900fddd24c716600036d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "f5018f63318fb1a850136d0be30a9d0b3ef231872d85bcd6316fa9a14e3635ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "3e5e907a30d05517e1d7c48f9a27b317441f1c01b40470f78faaa2d785a20152", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "46b70e68aa9a1f37946ca0a934d110d11ed6a10ca27382f9c676534b30547c62", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "32c00e09fcf0b3799a6d5d69d4fe51f2fb53b4f70fa849832b7102b3c7fe4e35", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 896, "008e47d375ae9913e6be253ba60a320d0669ce7b42ada5ec0fcbfa90f47d34e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 896, "60a937e90b89895eb6751525b3be1ca475c6e34357bdec82e0fbe5c22f8d2fa8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "7c1fd4c9d249ae7bb28ea5e20865c3442a7408d95089b05eeee78e1b6914ff9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "2021fa885b6bcd3b8debf9387e660502e9779c771d13435762afbb13f1511e25", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "893ca50a3ed16df5eaae7da9059f4ef39eca0bcd173570c085d3f4ccda4ebd74", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "2dab2c182e22296a2eed4336bb0562dee95c6d98c75d77912f3fabfc20af1a6d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "d5118c2cc34d746a67b0a0705be0aa0c700132ad185ab203c2c7c1bae7ee447c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "375aff1bfdcc43ea99638a53bb1f303b17ee24b8859d6a21bf57c8979bbe7683", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256u2_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 896, "de93ae374433645eeadfad24fa36a15c9882899745fea05189f8c8920bca1951", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "aeb136a0f0bed171f12cfa794a15cac961fe234d8ed07085c038410c60a5b8a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm107a", 512, "5551d57bc2fe92bc67bd0b233752fba9c2df4ab3db14283c332f5e137d3ea333", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm107a", 512, "94d620e7426002ab3af75c56fb9bc6ac737d1ab01882e5f8acf490b8e928c806", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "6bcbaf96193d433235a9b5bf0ad5234d48731b05ab0d5c4a31af6a1fea2fc26a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128u2_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 896, "d132d34a18abed57427eb032d49515895285c19b1fdc3c68dc3cd0fa660b50fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 896, "27992ce720a5d19722a7c36712c9c2b9bd535aa846c62c29ce8977d02bb2fca7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 896, "d032ad6208db0a81f69bc2ecc2950ab6ecf26b11428179aadc16133ac3e0c6f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "bedb628fbd3bdccb058bb2ebf167e8cc0a8903c20b116d97c43737a64196861e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "31666d940b2444f99ad70c41b0eb68cfa6fc9eb112313f973612e3c9e8f4cac7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm107a", 512, "1e1f853e0e2818e4c7f0ce6cfe2334c41646822db9882edc430c7ab17d7f7456", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256u2_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 896, "fc43fbd2b3c0575719ef1b979526bd6956aebb796287fa89539aa1dc77177005", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm107a", 512, "5c9bbdbce13d74e4157a505cd021624980fb71a1c2979b30f58b3a2768859b5d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128u2_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 896, "31a250f432a544cf04fbbc738e99ccf51a9248db8f031ec31eb8a471558a32cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "5852286b96e7b1050fc17f3ba9ba30d8202a7d106932d563822d989d30309957", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "fca178a780a79e68832af7ebfcf77be443fbf4d97eb76c81a50447e8131cd9b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW1_lsfbW1_dynB_sm107a", 512, "74ddf7cc092d3b486e354a01eb58c777bb9fe6fa7ef77bee5327c4b8086efe90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW1_lsfbW1_dynB_sm107a", 512, "a184f91826b9e481fdb80042fb07ca709efe45efbc7a663bbf296b713781330d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 896, "f28b62d4967830a9fcb1fddf32e3ce491a0a288daf810f7c35c9d5079109a6e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 896, "96d0715d8d63693ac0855fe5b9f9f4986b27c1778d72e99f5ff8e262f126e4dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 768, "94e81721c70b99c147ba93c765fb06954fd00cff5d8d32e0c3f5d35aa5d791e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 768, "04e1844b0cd31fc4644cfa8121029ea27d0a7fa0bf73c4cbeae8a69f0e87cb83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256u2_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 896, "723bc97675f3000b0f6469a9363211fe0df12391838e7d8dd7e8ca797c008bca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW1_lsfbW1_dynB_sm107a", 512, "548e53fa12fae524711a20205a1f76af46609d8e61b02baadd0bb75a7d8e9e20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128u2_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 896, "cac7162165f6682b8e4e28aa99b60e73c5e164d666c5c20579b96efc6b418c93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW1_lsfbW1_dynB_sm107a", 512, "b6c96ab75946c01ec35f4693bd41aed28c68dc3b678305edcf0caf7c65448b7e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 768, "dd91b7a760cbbe0913116aea21a211dfc1b455128e893b9bf0ac26b0d11dd7c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 768, "add310b109b93ae61a9536bd21c510c8c06ccad512efbd32567d89b6ac172862", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "d3ed1a016b9a2f7826cbaefa4fd1e7ead6e6d8e08d09d88e75abb575160d3fa5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "1e8f843fb72e6221e9c6d9292712572158f1dbb66899d62b13550e8fc1f850a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "8180e97e214c3c4bfff41898b61503bedd62611b700d4d08aeb3cc25041ff169", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 163992, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "3a0e1530d29cf159101b69bc863972052f3716c313c47fc7e88eb41578c8f2c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 183960, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "b83651e1f4d0544d76111a7edac381ab3b194dff650d17ebbb45f1f16592cce0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "cf8b5a545d16db132c59630e7db81b2a94e6a7f1f92616574c3d84e94c2a84bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 896, "50c83fac8fdd681ef99a0ac4c6fd8a2c04d23cab93df19531a5ef0a4a5410048", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 896, "f50e71da8d00ac14390a4f1369bc386378c32f914767f2c327ade627a68acf14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256u2_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 896, "aeeb186e9cb3172114fc6215b417960c692d1228d7fcedbde4a37e792abbb153", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128u2_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 896, "829bf36ad247f6befc9908fc636692f207845a90d47758e64cc1780c37cf27c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 191384, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s7_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "3c047c0ece18d75e11782e841b016f6f87517ade9c3ec6e181e8743a52754eaf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "b2061ce83499e4e15a149607b0a8e579efde699d6a3012ef36766f19cfa633b4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x256_s3_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 896, "5d32e852a38e5cafb43dcdfbe78138dce8742024852d3e265ed58b31074d151b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 162712, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x192x128_s5_et128x32_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 896, "47d9a120cb24cd4eec005ca5584e8d515275c88ee7b8dc8501805a3cf26b46e8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "050fa31471134fa8b106865316eeb0c35b54711518be55891dcb7fcbe2e07f9a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "53949d0fec8ea323b1c64df4eae0e5a0d52078b9a836e24fdf7b4dc709023492", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "aef03416003987a5cdd498881eab1bae9b615b3d2f1cced50098d03fa3c38428", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "3e1e0367d17024d35167e460d5c11f5573a4c2793fc555234e23353186c073dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm107a", 512, "6790c6ffe32b5e0e7acea6a2e13f0b2d7b0ae43486391f7616d17ae242975244", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm107a", 512, "8c4349814e91129f3eb7c78c9bda261f9c2dc67ae7338208bab9495f18eb8415", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "df4bfe8ab42b36812c699e86c9006d6ded527f4d4584e7e76af287fed246ba18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "3c16c646ef81a5655a7b25776cce31b46b63ca4caff47663f0d21a4081624f47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "e2e15f3d98dfae0477d90038e2986f1f785be2392a826e54ba1323488ecffb3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "d3893832e162ddd625318f2a07d4cf00f2201727965f58230ea328317b1cdcce", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm107a", 512, "24db4c990e7ca1604ed039e86e247202004a856d3a38bb53e25309c3b263056c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "b61a633997b29a4c374a4cc2fc3883ef3ebe263b2aec5a82d150589a29a42f69", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "56c7373c582a0a01398b799c4916606d0e55cb3a4df7228925b99e2dd030d152", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm107a", 512, "e96ae5bf874c30f5f6ecc0d0b10651d28bb78c6cb50e25454cb6bf6bfd875e21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "e67efb22461620d1d5001e2fb51b395456136ada036d497b8f119e07a9e1f791", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "0ab34da48272e2fecf420334512432a9f01164743f5cd6b80382761d9ec401a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "f7fa557fcf795dff46053b52c5777466a179fd1d33e2a48ea0c8c495f24219df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "34082c4cfa84cf5b98587b9583beaad12e0f1590e07decdaaa4aa4886decddfc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "292b296dd8542f3369731c4cb377373b46c235263a500a897daf4d6bd872866c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "38dddf024337ebbdd94b6d2d0cdbb7021016772457802f14fa2f01f5cf2e3014", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "ff7a1a2f4154bbc81c5f8df62a5e14905b7412e31f6e104a8d937985052f62c1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "b3728f67a2f63ef9921f38d441324f1283de8160302b369a41cdd73ca5bf898e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "111f526cf7e68740fa04861e0275bb31aa0f3d76d0ce7232e9b16faa15ab9dae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "81b4eff8a2f4bdd947c9aa7a724516ec6ddbfade4b02dec34ad869dce1d75039", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "4896c19ccfa6af8ad741bc11ecee839d594494fa58ec2bb96ce40a9d7531b5bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW1_lsfbW1_dynB_sm107a", 512, "23e51fca62ecdfa33ebfe946d749e3df26aaae5103c025b154b350c01c06382d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "046b4fb03078f0f6d0aab0a027565d207a1147313ea792571747a5501a6d97af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "43b312ffffae5be199623c1155610201aaab1987ee02f8f9ab03c821585fba89", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "b4b08491bf2de6be15bac201b58cdf7139c399857f89c16b1ef924ae3ad89915", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "aa35641d24a4e8f29556fc2d1d8e202705eb0825a60503ff72ae0c70cc40d8ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm107a", 512, "ea19e7ea028e5924b63bc93e3ee3dc4e373d9087d88af69ee86f6947cf51abef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm107a", 512, "26f3806c3dcbde243c07cfb51e91113d3204c7e91dccbbf6691a35ac9da11072", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "5ba6c2457f4fbfb16e9f451507d7854b9f44651333788a6da293812bf7e708af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 768, "201f90db6d10cd1ceb7760206da25bc892bcbb361e6be36e4241052b85f98919", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm107a", 512, "f023109426b25786fa5e26f5a15dfac20a4c4374528b2d8bc83ce631bdef1f55", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW1_lsfbW1_dynB_sm107a", 512, "a70752411f104c1fde958e2a9ff7412d1ff3a74816193276a7f4ce0ef63775a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "3c6a8bb93934aa9113e4cc4a22c3417b10b30c2c32083957653807e7bae83b94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "65597aa09e7cd31ce9d0138dc193195777652018653a4ffa1ea87b57435f1adf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256u2_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "b365ef6b777e83fd77b3b6a11bb09877751fef8c7be386f2429379fedd190200", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128u2_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "271a2483c47e2cc78ca6a34cb449568b8434270e3eab3f0859d3687c49b295a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173720, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x256_s3_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "60c1aef228077bb0c3489abcde1650f7eef6c12eb2b88431e9972aca3bfedef8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "64ae16a6d9df4cbc0668f03f4fb5cd77769475cadfe4f90831b04f9a7fa7280b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 173976, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x64x128_s6_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW1_lsfbW1_dynB_sm107a", 512, "03362911cd128e0a142183b41cdbe44340b68ed9d509710599b5aac973af94a1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 1
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 1
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 768, "5b6151eb341fc8d55ab0695a72cb213bacc7217ae30117962771f03c022a9047", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "0cf80a165c05bef1144e6d0c173ab7a75b05d4f4c5d2b6a40422108d21a09cfd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "43fa477d0e0ad206afe6ffb4c902993419208319b1207647caea6704e6eaff96", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "148a506a932ce19ef08704e8d75118d2b957eb19743900cd11da7946fc05a676", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "87365efc59ef2bd68170ab2042545c3abaeefb5c3b528fbf7e80fd6200bfbdb3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "348cc742b170250d2ac3e3baf755b20273873a0837abf7bba316ff9882d9a992", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "e7b8ab91bf767aa03c53dda7ff51005b6c122e9fb80e9ae2367aa325b2c4a0b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 177816, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s2_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "d9bbc0def33e7362f14b75dbd87c9b3707f51ce34ca2b3d434914a14fd3f1d3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s5_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 768, "9ffa6653c54173efb637c6eb56055bbd5320e08b9830398b8aab367767278531", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 152
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 48
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 48
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "fa06373db8e7db1005079371034e6e1819e482bcecf2a3bc838fb9371294ffec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "8eb6628835a7ee9574c645fc38a0d4f9f6a02094ae6b936f92503f4532340c29", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "1c8a54f8a6c5b35a0cb18b324198067c61b9e5aace6a6ab60eeaebf46aa774cc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "f6d4eaa34f2253db421534329c7a52c38a3c3164ba0392adf03743cf958bd5d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm107a", 384, "c989f19a46f48205e953b36f47a4e37be216e9bfcad12d8fa5ffa22d52f24bcc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm107a", 384, "0f9b476684feb5b6068e6d26cd20af651fe1eb2a5bbd5e77606b8bf78e728534", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm107a", 384, "314e30ac120ebd725ef3c3cc695a2747cba384cabdf9935d52f2e11f2c7dd428", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm107a", 384, "174f34bc8d6edbb567c2ebecfdb5480b76063c2f55e9295684702b520d4ef27b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "879d610988d48d9f28417d48376ea0b9eb6e8b7387b8533209579e630ddd7387", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "b0df1f9765e1e6f95912f50fe5d72e1fb8cbb08e213b8796e813ee58c63434e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "f6462ec690d073aff3d49b86250d92b7ec01005a91df5dda20419140f3bd3f3f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "1ee914e0844b1733825df568ddbd44d012ce5a324462c45d5881ffd265d8fc66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "00d8039985a35cc58c6896ee13dc70c9ebfc0dfa07cf67c2c368a9cec01e7b2f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "c600473db52031e5837c06f1add04e66fcf80c84f15cd07b3fb89f5499f05c84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "0857a40ba9a9830c0bc05a1bf530db84ce3d747e81500f2f73920c992f03d0e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "24a3c8fe2bf916daa49896658e00cf8f0b2cbe7ec1948b8759d5130937c370b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "05efaac91eda5e73e8ee172f94353a4e26ecf30a7d78a6497824719cace0680d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "4f9eead29543586142d39917aa53c08b67825bb4e7980e022b713c983f2e9a77", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "33d98cb3dc098009f86dc408a16ac6d91f42b326dd1c7cb3a644ba199cb0185c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "a8b4e92a2024847aded5109e0a492ef92c9efc8dc7be6ddd31c8214efeaa9f1e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm107a", 384, "5dc6f5db40561363339f1be903c274064d16e6099b262b10d2d0bf57b5a04cb9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm107a", 384, "3de8a52e793e21d5d402928ca7c8f03ceb32cd1a9bde5466b774b2b18aeeef4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm107a", 384, "5901118240e4d338b04162e51f747103272c1bbacf3cf7fbf59e52276dbbcb11", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_siTuGlu_lbW4_lsfbW4_dynB_sm107a", 384, "368333d64bb859d9dfaca8ff8a6f387d33643c802e635c6ac98ab193f75edb68", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "88dd6b6937749ec5093d70e5bf54beb8f380273dba59670a7df70be8fb87d9fc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "83c8cdcc85301b2554e30910d91e8771e9427f3a7f7a92f2a612f0cb46413b47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215704, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s4x4x4x4x1x4_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "4e65f6c47e17ec2c75ad5ff2efd9c05e9c89d2b54cbc1304162cba080429e8ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164760, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_relu2_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_lbW4_lsfbW4_dynB_sm107a", 384, "6d984763a8fdbfb05ad88cc6a4e8819eea66d95ed2e7af27df6bf82c989f8a00", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "ecc4d4832546985bcbd4f88ef4c8383bf0dadc0919482b22a1c8c13ed342d4bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "cf42e2b0406f8b60085567d89fe916eb591dab4da3c62113dacd10ddbdae3e83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "5020392ff3f77faf5219a687f2b392c44e52530bbdd41a60248f6b1932d68bdf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "414a105d46455423e76ff83c8e31450b7e556970e0936d7ed6a104ab969216ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "f455084fa0296085eaf690a6cb3f356aca28e4eb547129d60ebd7e3e37c77138", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "253fb2e812fe3f278721bc63031de757cc7de5ee49ac21f976227053f130599d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "8b9b58598e5adac914c0964f3c19dfbaebe001242f10e8c7e19be5aa06e1e9cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "83b019a01417358f355b58d8915333e574ee5d9ac247c57d28f5d0384276be54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "6f1c61bea1a2ccc2adda74621f07e5788d6d0104fbe75d47953825eed48fd225", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "58875a768483cf3042be17a9791a87e1d9b797e508571850dc3f930d28a6fc32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "3ad24077e8360fac9452dabf11054a58ccd5c417ce4c30149cd67f490b684c13", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_lbW4_lsfbW4_dynB_sm107a", 384, "e53a12e519809a31e381c215645369e54bfc46040679970b74a3c254894ad981", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128u2_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "137d8c546d9ed35cd0117f5c35827166e5ad41936b99eb1d342de0be660ed011", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256u2_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "60aa6db38ed55a3c8180b102acd2c87498ca258c9dc5432a7ae57c4261120e95", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x128_s6x6x6x6x1x6_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "1b9dccd3ac6d2bbb5b119969268436ebf198fa708e8609d9f5145479cc814770", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x128x256_s3x3x3x3x1x3_et128x32_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_fCp_bN_ldgsts_ldgstsSf_rgTma_clmp_geGlu_lbW4_lsfbW4_dynB_sm107a", 384, "2bc427f5de05666fee8db8875796a75af7ee645e8e449ba7372db1180c206670", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 1
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 128
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 4
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "b738a2e6cc20bf9919d7c02cf6fa9da9be7ab8f85b22ea095502f75206d4da28", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "4a27b2de8d7e3dcd810ad21c66f2a3c18aa4288f228da276bc3257f4dc747d1f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 640, "210cf28f872691e82b48d785f8ac319480fc92268f6dbdf1c43d856aa9586502", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 640, "52b03982236c8411cfddbdc3f11056c55d485d61ab40f25bfda1b2a62472f667", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "29c1dc8079519cb08d912f2fc572c0de2e8178a0ff6add399d61e9f869018fef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "7d3c6e06f248fbbee323c26bd18271efd7759e39efb7ffe6127a97fa96c56470", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 640, "51c2cf31e314794a055a9ad11fdb6e9ceae9624f16b3eaee28e40c0cf334edfd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 640, "0df0ead760ef77a3470e95f39a3b3dbb31d792804bddab5e3b18bf7092d594ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "69b582dbe60db0b599af4ab7e0034d19fe0deb31f058822fbe332d9eecf9d665", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 640, "3d7c640c92bea508a55cf43e0198293acaae7d4a6a1cf5af2054f054ed226a0c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 640, "24fbcfc7ab6f8d499f92c36c3f633215943c7dfea81c9bed1301a4f226ffb8f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "93fea11bd1b9807f161d1efa6a74fb4cabb4046114c09a4fecde1e0aeaf94057", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "2e3c8aaf1b7bc3b3a3427552bcc29bd75ab2d5eea45ec6230b4d32e4a45a3cab", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s6x6x6x6x1x6_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 640, "1be46c221c9c7d1ce0a31682b22cd02c4204fa933541b8093881dd6fd793fa7c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_siTuGlu_lbW8_lsfbW4_dynB_sm107a", 640, "dd82f3f24e560c3b4338202b36307e5d3f98b40249e4f310397ae85860374f9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(2)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222872, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s3x3x3x3x1x3_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "2ff08515e9bf7e066dbe2d45b265ad56a061892ecf7016d7d27fa1db4d10c162", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "af0e4fa3fcb6acd041a230ae141642078546decb2379d0b29b7db87535c26ebb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a369523c932a4b329a03d9558d2037db9d278e6ed080cb09b13715c39814b68f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "9d4a1dcc64dd9532083825154e9d06e6b8a6160cf35eba1b9482cb0d08479ea1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115216, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ff9680d5660e388528c2ce8eb46979b8599e33589a31c51791ffb29ad47948b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "3b38b7027341b8e1ab10644a75798e8947b070b7f4424af9a1dccf7b8076f2ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115344, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "83c08570c0bcf35315745e04106066a42928e802005fec24bf48f264970aad72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151056, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7bd7676fd77b50b14b46f212fe4d2f187617d753cfa3e541caa678ec26730505", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ad94d536d90f1bb594a638d712556656239a4065978d12ccc5654686b1d193f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "abe3b68d24b88b03fd7a1a13c4a37d4fbb77c1359e76d1cb97a2dec98de02387", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187280, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "829ce087781a9ee790c4cd02c081fd10b9e70c6c555f68271e371734712d67ca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222992, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "07ef07532b98ec7f2145fb8680e3048a9a2b695d344f7879d8b3903c48a6a909", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223120, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ece53c9601476d97e22b9a3f9a7159f3829f849b06f621df269d4c393a6a79fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123408, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7e6fe6553d8d6b34e4eaf1c12ce8ad414ea9e914c0db85559cc86a0a29c1f727", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161296, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e23f9d834e9466de7472d2084b198673814b13161ee6d21136d045e2c708723f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123536, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "781f280ccc8d5e543acf2efda263d64b64b18cb4abe62f46f7d74a023a6c3545", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s2x2x2x2x1x2_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "523637124726bcf1734268bfb0a6de84c0d9bcb45ced7d55df92c23d00f6a9e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "04b4c5fecd3da2b1f4e327a86a76c3bd7514d2ed911892b215c06d8e28a5660b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161424, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cf8cb0a5388b28f1a39499828f2d5ba609e2438ac910e9e0881bb3751297eb01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1f390caf8a2ba99895a41366ce1e604045eb277226c2abd50d58523aa8955904", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s2x2x2x2x1x2_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "ed2fb11eeb3bcd7d8a6ad4a46a15ef0878cdf506a4331ea4a36807992695147c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199568, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1adecba401ec518d3db6fc97e1e2084fb3b6869d25523619dcc870fb867b7e2c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c260daaf90ec91334dadfae962ab0742778a453281a58db4400f1b58755e3ee9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_relu2_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_lbW8_lsfbW4_dynB_sm107a", 640, "4ca721a21259655c3756294053f58cee5fe76923682adddaab4ce4c8611b3876", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(3)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s2x2x2x2x1x2_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 640, "0409e79b7fa00ea1360cae341d1390bf0d4fc1df2cd9a2ed8d5f03841083aa4a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117400, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "9c3c56b19c0b5a3eaa56a14ef65f9675d6e845f83ff827ab77f881d87acf9ba7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s2x2x2x2x1x2_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "d0bff25752f6c81dd43d738b5d75729e21eebebb4a8f4756b8a6494acedc2044", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "f5c349aadbd1c97fec3bdf925e76f06ead585dc66cd02cd2db979d69695de292", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE2m1MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 640, "43b6b2bf30148fea2153572b23b1f186e91e70c1ef580ab8481df18457e68adc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s2x2x2x2x1x2_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "ad22c1165f7b4d3c50508771fe8f7482bad1071c120c8ab5b2b7e94a91c1f888", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_swiGlu_lbW8_lsfbW4_dynB_sm107a", 640, "62490cfdb09c215fe16ab349b09b05602b76238348ee057a7ad47ccd2c7f37fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x128_s2x2x2x2x1x2_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 640, "e2ce820e8eb49216609d7ad5ed779c665d7279041a81403ac036d3e0853748a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215448, "bmm_MxE4m3_MxE4m3MxE4m3_Fp32_Ab32_Bb32_Cb32_t128x256x256_s1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasBfloat16Mn_fCp_tmOv_bN_tma_ldgstsSf_rgTma_clmp_geGlu_lbW8_lsfbW4_dynB_sm107a", 640, "acee16b85ee41ff565b1ab415fe4ddab9f85aaaa63422c50eb93ca10826fac72", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(2063)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 88
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 1
, /* mNumStagesB */ 1
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 1
, /* mNumStagesSmemSfB */ 1
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ 32
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(1)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 56
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 56
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 8
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 4
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "68103541492ca9c1fe4f29be7bc3d002b49f38a31f42e1e1d73c8d8083b42166", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "22cda8ff694a2a9e3d6178df11d1d3ad32327514be6afeba125b0238a327d92c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153240, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "5d3b9226583baf6299b99c4cafe52d5d4de26c15421133aa18de0a6fb82c6200", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189336, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "88f170a9c95d0d7840e1d1721f24aa7e00c906feaad0966857421a50ab935f19", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225048, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "5d531d2812e9a68608fcb2a41b6289804389533b0df228e040de345819d00292", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225176, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "27b91bc969821de9320d0ddb29496645e810c1a6b292146d3b5c42f0c453d7a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139792, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6159616b04958a3834a36c8420779ff36fb94c3fd5500b4bcc95c4b08c640ca7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139920, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0ee9b08528d79bc8e8ca26bf3fc44d7950a84031b2c6941cec9f3956e7cd0a4e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181776, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "64cc32fae5b385fa73c40475177df07fa1c6dc9e2d41bbf0c0b0277a7abd9fa6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181904, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a97a3011a9a2b170ae1d4c5f7bf5551847ceb103120f44383b245c44c16a7257", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224784, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2133bf66b4b4fe1974220b74987a260b01368b9c1df5e25bd396725dc6d61567", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225168, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b796339e41e9951a5ff259c2224234a0d784904ba8e52bbd55704343fa0339df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127512, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6d43e0f98511b60ec6b9ae2de958e3f9726aa1e0feb6a712479350740ec7b44a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165400, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c39e37ac5fa6b39ed3a533d226db5c4370ad647e3befc9c084bb4f2561dfa957", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127640, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f1df2c0b48e1ad61dd24e19e70645b76431b44ead0a1a456aebfbb56cefbf306", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "748129e2d071a5689b340d35a35fec43ccfcd00fcfe5fd336a57e24fb53c6761", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204312, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7045e8823f04b53b8f0da698e82474eb03b1b1282039471d98c0631dc2e64176", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165528, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8597e961b9dd10c1abd1c4125438b101a2e18629f01f6db8dec7a48d70a621fa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8e7a8b54e326df1f6d04cc88ae8e115146dfc8c727e131343c0cda7168211ffc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "bc2c4f074883cdc2385b34adb48f381b326d8a57ac6339d5ac7bbe509cfddcd2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "883bd1a04122fc36276dbcd62ded7978f46fb6466535b4145c95d32d29b15b4f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "913230a797145c2848bb8d0d3447209e27686bea9a806cd0029944e45f7e6b2e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a167bc1b3043bd0704b0965e487833e27ae660eb0d28673197855776bd4d05bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "547bfa1f0bb67e340e1551656535616f9adafbc80d554519b08f58c53860b181", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "aecddc06e4355c2962a1f69ff9de6c4f9bec5f5de1d36d609fe2aac765a3ad21", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2791b39b6d758a26fa28084de2a2bdebd1e224406b238f2db2b53f26c839f940", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "198b4e82882a6f46a90211ffba177b678f4f922f04496268c74759b4a9574598", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "9159494ccf6f25a6bfb055d57014365873ccc2e744708ef8060f6a342023b502", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6dd5d0d8e99c4d086d5e5d85a413dcd24448a20c834df5ad6432008a7a944f3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115216, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d8757d9de3fcbbf363c60027c5816b6ff464970c3ed298504a79f40a8d104291", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "75030923187c90be6eb4fea2ed16568cbb02bdb0b35f8fea6df9cf7e894b7d7f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115344, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "06abe2273f6693461593a544e9c42b417f875cd1e265249e0f47e959b58d5c14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151056, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a4a9d9d5d669b79c00e2e08db458b49ebee71af484f4d8d2e7cd7629645095d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2ed224ee937082a33f076f998412799e3401b41aa936a9925aa6245deddc4214", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "dfef7ee3d979499d43f917b435a9ccbff071824890ce866234d7cb2a8dbd6a65", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222992, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e3b9490f0697dcc427b0c31bc24aa8ca24e57d7e016e8cd93bae29d811ad7ce4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187280, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a91babf900642c14121dcbe3b74e707e19ca7af46e18f790c2151040b7341e93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223120, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c327ef084d506860b21da6253ce3c5c9c5c59d4d8095cfe0ba45cc308feca651", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123408, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "bd598a76a6099e07c7f863681e008531b8b82de36b525d4dfdf9a976cdcc052c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161296, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "28b4d427ceedd6019e0dffbedb39a7d52a97b16f97194e770b3ac84db080cca5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123536, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "af21cf0b7955b3b9f3dc42e9012b61924b1b3d10d8625626c885a8373e8a7b84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "be29c8fa3abbb4cd7a0f387bb9ad0c6912642864cc70e2ad45a355e23a418df5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161424, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "4c2e04207f5af9f747116cbcb81e4dfcbe3a554bf14dc9d950c51dde3fc201c6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199568, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b17b41dd12df950708e1dd9a9a804df7803cf2d9e1025c9acc71c7aabf6df960", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cbcb8256520a3718d15ed865a63a6cc076c136efb21e44a983e687b783e356bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "fbb14066307e8c82cc09c60fb6c95d7903e3bc5e92694c993f41e57bf5793c70", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117400, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cbac6a7df19edfbf223ec8213f324d7c50fb0a3647e6f493e4b9021d769feb61", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "9806f42bd44c3f6c3ead0bec5e911818e0dac6d25b5c46a5616242468e8f9654", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189336, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f853d53ac8173a51289d703f6552ce1225839b3dd0372f216c017a2bd4a6eb94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153240, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2ec0df29c4bd2ef5efb99825164c1f6dbb599e05b6b202d2b7e3564cb252f278", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225048, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b011a7789628d5e142b5657a94a946d6797a3d481efdb275180c9a555c52c62b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225176, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f3f20c91d5c17d475b17a420b001a7d29c05345759401e4802801fed9a7eb368", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139792, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ed5c614616356804f90bdbf596704959a718b10d95604263d70640c176695089", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181776, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ae4908c5cf9c20b11217d00bb3c899f1cc702e0dd43537fdb3b5c6e7ddd6ed58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139920, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "158d18c6e6f7f59c87022005884838b825fc4085825efcdbc97965e001cf5fe9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224784, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2c0e5d4dddadaf658b0994b9b46ac462aa1c75dc4f49339487b0f1189625487d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181904, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "aeff9266f761d72893a619fe8159287d6e4d0e8e110d7b65270e903acf89abd9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225168, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0184b32bc126738dde2867268e3c30c390843148b4cf5ab843e07a5be98c7b33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127512, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "99ee0aa90abca7b2351626365ddca953cc85061a3e2d1b17a5f778cb18d549a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165400, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2ef4593ad698bc6b971efc72986d6dd62d6721cea3b32923a06cef2d22c45a79", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127640, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "145688a3236aa7c408239718c33bce9b625bb54933dc82009da2506cec0a131f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165528, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "58eb3b307ddf4ec9f77813dfd624daa02584ea38f800dcef62a45aa32746d285", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204312, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "5df734c2ac64d51de286a4a646fbeb60b37b65067bd90f491321bdc6c3ad1f9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cb561a660d22dd050c48dec31f756b9a26e4b36c57d804a2edaef36cf1308142", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1233383aad1df2a1221fa1b9746922938fd2a8894f0dded16870dd75e67c23b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "553cff2d78a5eb2fadde76089d832cda2a2270704ddb55c0cc28fde4026ea86c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f8cd1d2c1f28456c55f2bc2bc105632d41af9177f4bd7864e2d1a98712d53a43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cf30ca39e6125fc2d1abcf61eb68c29d15c3999e91c9958ea074d2d2bf49bd98", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ae4c5866d08da163d9b64d0b4a984355a231e0194bb1456aaefde9db26f92325", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e3d246404cbcd6777fcb641937b685bd84da17a868330388983ff5412ef72abd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f915029ad934b41e3e54b157f196878d23e61726eef8e756b221f3fca023d6cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "72f2b933029291876a76bba6f288156a1f2b54396a367c36c1d2199ff300ff3c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "59bab3fc2298468086363e662572666e4593a47ffd2f80da51d933a28ab6fd90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "743113356b5bc5f6bb0ecca56c862847ed8bb2b8418ea24753cce2e8caeb183f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6ae35c6e9cdc77d7da4c67242ae43cd987f8ccf2d5532ad82b914f501d24ab0a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115216, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6c5ab22e7d0192709e14fa20a80ee338e7906c897b1cec6c595368e1e2388611", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151056, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8c44c622d8df1abe2707c65dccb63bee1446a273fae4c1404e29bd0670a04480", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6943eaad23e55143a30c395cde9d314080a11b6f5a8d4a2dad381c0dff5cc4c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 2048
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 2048
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115344, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ce11c0541700a60a895b34e7c532ddba778539ec8dc19bb291dbe72d5937fc06", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "dd7dc74b4dced54342e9f06c1e61b1d25bf192f4cb080f2f15f045ea39c9b586", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222992, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cde76377b3afa523fe43be5c17f471138240811b4537e4df1730908b022310bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c240ebd08dc6dc384adb454eb663eee1f3dc66e7af59cd89018de404ebe0f660", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187280, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "35d5ab532d8862ecf744d0d25ee5c6efe9155028641436312913fea7bbd104b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223120, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6d80d20018800ff24b812081d5f1cec0722969c04f3eb82237a7f632d9942804", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123408, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e09598fbe94d98ac980d98dab71dd7c3689f1f4ceddbb3308d6be7626ae85518", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161296, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2d14da23100e8407f3bbd69d2235f5ae24244c281f95f14484c318ddf739bf3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123536, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7297ef5ce88fd5c944771cee3c7903cbc57f6bdd2c4a6c7c32fcd8e8e400ddc6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161424, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1f00119d61672f9b0362956ed4693bb65148d5b9192844a4c86b2994b5548680", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ba399ecf683bf90c6a98160efcc56f0656d72de7745d21db6ac19770559922f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ef7786d305a918770c3e73d56bd73e332ed721795213f8a20baac167f3c64821", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199568, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "5af617a2b3ebf357cc167106c268346e50ffafd1cfd31627c3d350ab7fdcc8e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "3e38bf077fea147fd65287b2a610826d451f765b0b11fadbaedb4077207b4ddb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117400, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "3d2c6b6411603225d253e3941f43af70bfa5dccdc62af7e86d22ace045f034dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153240, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c99ecb87427d8ad6e05e511c76dd0a293ce293289127bece65512e5b78430e6f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "39abcef82a894eca5372b378ba5b1e5df0a51856b19702061aa65f3c5c38f176", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189336, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b37f7b8f4f3b56509050943ec49da608adbafe79be9cd26e86f36d522dda9171", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225048, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "fba6b552748fd0ba6dda7573b006448d422247cd1089c7d8f375a29a3c0a231d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139792, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "eac6e9868440f6083597c1102fce4d33631d78297b30f43891651d13348ab11b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225176, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256u2_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "749ac0cd49de6885b557551a92b565ff967fa15ac81f0cabde09e5910e107959", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181776, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6043e85d66ed5f13f938315354b88e97a473675584ac53603b2736c038596deb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139920, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "3fc95cc78086d7c8f7ceea257e250b4fd90f3e5b7149c693a0102ece3ce8d130", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224784, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e13f36361a58b31578b869717eeeab69d1d87c04183fc60168bdb654be062bcb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181904, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cea0553a610f36a06aae9d1289d11efaf9e377f3ab32924f6b8faf9cb3f41716", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225168, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "bde286569370db75c0fed1c0390bc6ab7a7fbb89ca83587a4b655cfacb5beb83", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127512, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d223e16c5286efbeb875f595913e501a379fc40e4b2123a1983d4943016da66f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165400, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "bb617b43a94472246a6cd21cc2f769988abdb7139f5c224945c969c67f1c0466", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127640, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b36363c9f2f8ad3e6c5065c7e590a9fee819a7add6c05093d97d4dac87e612a5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165528, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ca38aec7af6ca7decdb7211ecabc3ffd1b180e0896f91b1643472fdf690c844d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204312, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a50a5dee7f5bd2ce5ced06d5412e7bd650420f31a530c7e5728a2fe19b435d6a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ed263d64aade0c45f7e905f88ac4239a633be9b606c0348d14433037c8c3fc9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256u2_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "25071a8e93a40a9328ebdf5f9c06401294a80e64bbaa54f6b81b8f59671bc0ff", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ae47aaf841144c7139cfcfd2de7552c7a02198896eb292c4ac8042af26387879", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "5889383688834c3c79d1b210c9d7a6552e1207e335bf9462b3b50c82dfed4c43", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d433d06cce17629b7a716049f289205c62c1eb69cf7e4a70f969dbd1fbb7c87e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f7b52854c3e451d209bace13cf1978a05ddbac0fdc2cc967a40bcdf246b28f67", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ac38a8ee16778fc42a8be483a0209f732cab6e756c7d2c0370e4287cb6b5cf9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222736, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b6aabe6678749f3afd5953c4915bf792cc3b5c85f134d040f24871067f07848a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "b51b4ac8785da5ba5f2bcf441c28f26ac94e5b80213d329c3cc8b72429ebd1f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256u2_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d120ec0bcebf83f1b3c44f6b4a4ad0d45cd4d6ed24dbe2afc90b378bfa3ca747", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115216, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "eed2fe0578d8d41f51a204b0333a052f448baf8d0af969f66f73c8e5b7b84421", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220696, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d5c81522812590ab948587a633947bf462e405fcaac7bac35eafa9df75d67c42", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222864, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "20382b4aa6ef5bf86e983af1e6a33c4483156ddb480f10bfe6ae77524581e7b0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151056, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0874825ece4a57d49265db50ad9fc9ac3e57d6cb0afea88e12456ec032b57567", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229016, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x512_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "3baba3e12160aabeabf77e6c6e5e45e579363478d713d16650fed6e5b4ea0bc5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 1024
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 512
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 512
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 1024
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115344, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cc88e5644d36d2e7091d0ab58b51a969bd425c9d1121f621698015a535ec1e93", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186896, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7dc0baa4e393c43807989acf3d808b684ed5567c1fe67f2f37528391995ca21f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "4deeb5f4bb7129cca9d7d4d886705027edf4f51b8ea8157cac8b2f5b7c16c5f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 222992, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1bad82eb99ea2b9d6d043bce5319739e6032ec4b5fa7cf7fbe29178bfb98eaf3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 187280, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c3fcf92a00ab29a778b446965b4c70eb7d15c18a279efb2c23726d970bed2b97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123408, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c53ba295707414e1e522e434953d7a592631796e7252b4b94509aff8c73a6085", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161296, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7650580951599ef529fb72e19999887bc9653265b9636756c7665e2589c35f57", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 223120, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d9eb374c0169a1285d613502b6c0439060d201e107d52be0c5acada2757847b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123536, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "76c7df3d0d861569de45009bd42386174606dbada2fbeb9110a59e3511b2272a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "88d01ce9698261d1e8e225488a34b3e31bc3f350eb0f697e882a2618bbb1cda1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 161424, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "de4c4fe8400dace17f1bf64c1b28e089a5cf3bf309bd748633d057fb9ddf5c9f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117272, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "77173c199d8dbcacca7e3f23d61b59971f446cf9237bb921a37e36e0115f02a4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 199568, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m128x16x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "4c56bfcd73a19bf3d3515cde13b1dd6c38f0d433e366a1ae259b644c6fe7837c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153112, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f4f34174216a2a1fc777cd85b1df338c14e7f62795f8ae7b21a6af9568753ffc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117400, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s3_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "51a4790172c033abf26e08a2c85324999b40aa8f12154b2a22be3e89047ab6cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 188952, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0e1eb68ff78bc77bff2779518d3c3843c642529e0d31f3ed7a55a961d02e1dca", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 153240, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s4_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "48cfd8ea5bb2ac5484870d61c46ac67b4f699d0729ab84b8d0e2e31a2c495176", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225048, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c6ca48fc7410792e583f8610739d35794d095b83a68dd2211bb0b642585106a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189336, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s5_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e283208f2d9def4a65bd49b389e89759d48b64c671b421bf7bbe0d26f84d99c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139792, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8119ea393e5507533fa61c4087494db325463809076c030f9f8cf283e8940286", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225176, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x16x256_s6_et128x16_m256x16x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "05d99e1f44a2ef2d2c6a65b5c46530b78905e4debf1078d1b82b38c8a7cc0c2e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181776, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8530d969752fcab544c589b385fc7488792d6686a1e0f2e2789fc2c2ae34ef0f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 139920, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "6e8f01ce152f4d20ff9984dfd0e493ae40a564022d0daf65cacb262baa9c9cb5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 181904, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a0788a22aac1339ace53fdcb22c381673bdcab24163f86cb27195f448a8de638", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 224784, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7df1b876df65284937dea61fc81a8e8a43e77b8a8b12751e01d245d501368b38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225168, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m128x32x32_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "a9c8eb97f99fc560decb9438d6468fde6d04c12e9c934f9327b988ee5f5b895b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127512, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e913c47a14ba2008c84f262afd4858c0ec02d2d2a4d68b955b57288102dc36d6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165400, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "2b808488c3ec02caea60ee0012d8ca36d771d24486e1081ee0bb00a0a15aef6f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127640, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s3_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "94f653581efcf506e5dbcb0e47f5f06827c4cb28a0df39d968b4ff5cf4fd67bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204312, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f53a12b7a20b596bfa487500d884b5d3a4351186e440f5679af7bee2ba365ce9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 165528, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s4_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d1c8ccdc2b3a3e818fb47a717b776901f1b7ef6b245906d8999935954ad2f728", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113176, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e44b1a0a5bded3bf385e44cdab7332aff3ed9358c1cc7016ef55a3cf91f433b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204696, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x32x256_s5_et128x32_m256x32x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "0084f012264bd1217415f77585107d2cefc089f295f8030174fe689c2d1af224", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149016, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "17957f6790838b349fc0e63cb156dd97c9bcad194c6958818771a8aca7937987", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 121496, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s3_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ef7df7ae37935a0f62dba4ed7db45b21099444949e1b5179433fd6c48be21ae4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184856, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "1acf778928aac98faae886d2029529bb6173cdcc19e0060dc8232518f2b4733d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157336, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s4_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d508d12666a77157e430ec9fab9a6c4627901b49fec8ed8f3c1893f780ae85e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 220952, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "4e82070f6c03add0e53d65df27fd51109bb105676b880cefef67ba68ca23ccb6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193432, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s5_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "cf3bf481fccc8cec1f91a6d1903492d169ef60239dcb50656fd27e1d4753c440", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229272, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x8x256_s6_et128x8_m128x8x32_c1x1x2_rM_splitK2_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f14b3debd743000e8acc0d228b36c895608a08b59d862ccf1e3466e3249b6ceb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 2
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 192
, /* mNumRegsPerThreadNonEpilogueWarp */ 48
, /* mNumSlicesForSplitK */ 2
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(2)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "5463ccc14d36145cf531caca72314c6242a3cb1e7603ea621540e0f7f6ba5070", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "809d06d945761f7cb92ca112f77e04f37a032a52b1c7d8a7a77951c6988ac9ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227992, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x128x256u2_s4_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "9401dec53d08111f5c14433687950c9d1f67bc258954e260de3baa330df79f4a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x128x128u2_s7_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "f095c251e3acf0c25520ccf45c52b302cb391e0b53757cfcd7c9215c22513c3b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204440, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x192x256u2_s3_et128x64_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "96d3ffc522fe5e118b79b7207049080053a24042ee988be4fbb50668ba9174c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 175000, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x192x128u2_s5_et128x64_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "fd682ea49db6e57ec6bca2f13ee040286b1109709a181457124ab02ccc4b47db", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d0df1b2bbeb134a3e5756e465d4067c549fbfb36459aa6820cc79496dc4d9ccd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172184, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "3cf5e4b882c1bcc2dbfd622f4a689d7a6992f9abd4b68352cb3a1d9ea21c8312", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x128x128_s7_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "f6ec187579599fa6d00ec85f646990c0a104b7ecbcafd51833ab0a13caf05f20", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227992, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x128x256_s4_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "eb937e829b75888118d67f9dc54313a56c918909f33577197474dd286b1e6f78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204440, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x192x256_s3_et128x64_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "d02f20edabc2a34ec1635650e2532ffec01a1c649fb8200904384b847344e0f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x64x256u2_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "7e13f27f6105499aafe0db3a40c322188f9c014279ced7b73fe96aa49c9b79ad", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x64x128u2_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c71d25de1c168a8e87530e5064005caf2ffba77276459742f5ac4f2f5a0d17bd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 175000, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x192x128_s5_et128x64_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "16682322a822cc786b5e6900d7e5b4e0e9052b1afc4d3f0fbbc83f072bf169b1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227992, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x128x256u2_s4_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "58bed9b29c258f17d5321316b21a6312543503556bdc61b26802274166aef1f9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x128x128u2_s7_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "85fd4f7ce7b2c444268c272aecfa8aa2c5355dae5d8b0618a5c254f45eff3599", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204440, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x192x256u2_s3_et128x64_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ed2d337170c506a42260f07a6f42fa4baa7960fb9ec3a9d2108642534f228f2b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192152, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x64x256_s4_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "5e3f8470200d9568bb78aacc3945ee91291ca2120ed89947a9045061e6e1be18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 175000, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x192x128u2_s5_et128x64_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f29afa4741d5c72e2fea946d97d406bdf0d1f7fb8ae935ca62bac085afa5114f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172184, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x64x128_s7_et128x64_m256x64x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "28184e42fcb6b7aec8eb7222c331e5911a91c88ca9cd122bd35c478fdc655f14", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 203672, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x128x128_s7_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "09d9515c17c820a6d6aa592175da854930e9c0a0a62d002b604589195cffaa4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 7
, /* mNumStagesB */ 7
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 7
, /* mNumStagesSmemSfB */ 7
, /* mNumStagesTmemSfA */ 7
, /* mNumStagesTmemSfB */ 7
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 227992, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x128x256_s4_et128x64_m256x128x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "8efd47b43390a8c39161f216a32ad0b223373265e941488c6f6ab1b3eb9e6de9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204440, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x192x256_s3_et128x64_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "fa98e9b4671a843f7c4727d8543908914b044d845814bcc2ccc0ab5d6e8a15c7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 175000, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x192x128_s5_et128x64_m256x192x32_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "35dfcc8d8e3be5875200cd48be539487f586a76ebf40fdb98c8b7b1ee3777255", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 192
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(1)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 192
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197272, "bmm_Bfloat16_MxE2m1MxE4m3_Fp32_Ab32_Bb32_t128x256x128_s5x5x5x5x1x1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm107a", 384, "6985ed51c3254edb5e2385f7412fff82e6a740c046c9ae4fae1368b6d59a33ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17826828)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 197272, "bmm_Bfloat16_MxE4m3MxE4m3_Fp32_Ab32_Bb32_t128x256x128_s5x5x5x5x1x1_et128x64_m256x256x32_c2x1x1_rM_TN_transOut_schPdx3_biasFp32M_fCp_tmOv_bN_rgTma_clmp_dynB_sm107a", 384, "b1675cfa6ba5614a4e6339029f50cb1c6c88ad89e4c8f939cc729a51fdfed726", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17827853)
, /* mDtypeB */ trtllm::gen::Dtype(17827853)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(17827853)
, /* mDtypeMmaB */ trtllm::gen::Dtype(17827853)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 1
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 32
, /* mMmaKind */ trtllm::gen::MmaKind(5)
, /* mMmaM */ 256
, /* mMmaN */ 256
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 144
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 1
, /* mNumStagesTmemSfB */ 1
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ 32
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 256
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 1
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "346780e2e20319fe3203a12892a01f7b94037167b9b0a0e845621c0a4e9be858", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3639fbf5ed244f4914eb415a99b761081216732c1428ad5fc9206ac8aab698d0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "383659c82e228b98d6ea72bb1835aef3609497f29c2f60aa143c51065ceb0027", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "38b4d1380cc96265563ae437e67d9741b70c2c0ba8e94f965c45660fbe1c6157", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "67299cd1d9e05c15d166b276062ca214f80043c5b49c51e852541fa1fdc1c9f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b1bbd5e843a56f95530b573fe14dc5fc283974b91b0ac79dbe8bdc752cae0ef6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c3ab97f600a2bf106695c7c66e56ea359a5439967f2230bd5300bfef81665ea7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 72984, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e9a3141074d3a61f814e8de27e2b63219c5d66a24220309c21baae1e9e9c1026", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172688, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2637a286e85e3d5c94bf1ca53d9455f611d03050864b6943c01ac7efb1d44e54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 73368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c79ccc67583b37b15a68d85e2010b49b23fc34bd5ca53324b298743d7010046d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7afb32e832d29c10683fb5b9495e93deedb363893d7b4a8dd22bdfec8dfe765d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2e0c5d7d20d8a0e32a6f7b4177c5ad03a1075dd8d391e099c4fc0337a941b763", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "694a08ab9aa514ade1acc13256f569ef847ac4f24c4e69a465a6d43553d9e0c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "169bf34799bb42da5079a6742769ff6822473dc1268b33bc657567c266d4f9e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9aed33c9379fec8a683367fe64108780253b0660cd3960d5a65265f03f0b1b9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a454ab7fa492102a996af8e00e325ece4188454c15d31cf7609dc8362ee76e48", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "83f71ce6948988d5918efa0fa0c107ed7b147cd35da15861dbc6b6b892c8c6dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "953d19e44a79551ac86fea42da68d0b5f7c778aa1f99c09b52663d88e69755a0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a349bf80067c78ba66864a69ab5479a7bb467a8ff3652d090141762aeea7bb34", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b7113aeeb9142dc8fac5fb3d716d7692d2b3940a1d8ffb26a5fcf1e9101d2435", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1651dd2ebcf9e47b6ce224827a61de9923c56f1eb590fe7242756d92889b99f7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 72984, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "16c4308288f47930449038932c8ddd889cce7574911af09941e7d1e7065b5cf1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172688, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "de11fd0763455dbdbcd5c488b3c929d692ce1e770167bdf3582702ef792b89a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 73368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7a82dd118dd1cba065dec287c4c84c480efa54331d8705f9335f52c74f94f657", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4f3c5a754b7239aee0c64bdfc90879143e9567b9070556b11d67fc66d4250c38", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6261d6eb43364ede1623190c6a459f38d3eecc35af78d3049c6d71dd38c224fb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f5a6ff53f06696a6cb690c4f0c052e448184088398412f2d394671ac743756cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 75024, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "60a6732b38d333fcc7326f5b77307863e6a57f77abf79316b82ed85a6b705b40", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 75408, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8d4b3d580ac252a64d45a311addd864857f5d10f6ef9192f6c56f5e1b4bfe8a2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "266ea158d9e648364fae86ed493e80091a7eb63ffeff8fd11c79757cb0179584", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 93456, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "34eccf8b6c84a2c614115a468f090c4a389c8877fd1879c298b5845d6fc8a8ac", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 93840, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "aab6d5cb0a0b58453961f3786338d6f34703f621ad8da03b4991d0519c90bd66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 130320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0d45afc6e133b9873bc7e44d48310fc17067036b15db0ea57a820336604a138e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 130704, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bb5394e9211616d2b55e3495296cec7a7f382eeb4c89ae75537d2299c47cab47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204048, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "31fd4ac58fc1f0763615c1f5b532864b683bc474511095db17c08587d90b7c9c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81176, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "45da57f640f36f6615710fef302a1abdef457bcee0ac20f318095ebb375bf7f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81560, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "280d754fdadc4a7649ca348585f72a969d8c09e362c74601ea0ce707f4160151", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 105752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "561cefa16dcfac2bcaefa5a8e4866dcc8bb1aa4896027c046707e1bed11f20d8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204432, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9709678286af22fdcfc0e9727084afc1542d23d37642a6db635dc491a8c836dd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 106136, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "35178fd77a8d696207e969e4413cf2878cca7f508cd8b58f8e07f0c7d75786ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 75024, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "279ddc09ec3e8399820fe736ee4940cc5ff9ec25abe92d98d659471713f96dbf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154904, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e692ef445119697db498e33c4f19bfc66901ec0186816ca98591547ce8644892", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 93456, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e0f655a6f82f582d4ca2f59b779874807d608b462d3c11387a27e3ef6ea38e71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 75408, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3ef04420e112dd0f29f423050e023d50749579e518e4d0bc25bc06ab81a91661", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 93840, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1f2829a8cb83b397da1cbdeb096b761d2a1c1dad2992ac80c02fe735d25077de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 130320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c47276008e77f25b0a699093bc77873f58cd3248183468c4f484218db4689426", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 155288, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7c17e7e0c3e02c739715233f08cb3b4b8ec6977a30f8d331dc76ef7002cf2fd2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81176, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "01b23789ad8790fcae3557a1749bdeffff0c7dc4618bff810da4eda1e359fa1b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 130704, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f9dc4bf198a4db4940a65acecf1fd01ec307d56c9f8bf26d74b64fca43fe5ca1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 81560, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a5ba11b43091b0eb462a93d8f76d4bd4530c7c18d9e42957d32f2d202a921606", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204048, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0fdd02fe05b135c5f2334bfa5487fbd736d55f16903faa0fdc0a2c4f0ae2bba4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 105752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c37cb1bf9f7ef573886105f76e388880c8acbcec89b22e311c71a6fbcaa4ec5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 106136, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "0841d1ba07fc4a72ab871289bef879fcd99ed160a49fae7be72e0012f927dd66", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204432, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "cf95a86f405ff70ba8f3fd486a9c2d63d4b300096f4091abe60de11b05494368", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 154904, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "73167bacf61676a62464a4009325047bf5e8895e1104f67d0f484336997269af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256u2_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8a624d44ff924a5d34e6841e5f04079a547f234dc4fa480ede8ef5606551a8d3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "d881d770e0a72999782a1ef203da32ae1998a275f3e1b9beba7a496e1cafb4d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "afc8dcfb693d5173089f49dd5738f0690b1f25d9db91d91c3de2804b54b9938e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "8a5c7b7b3c5a9071ec2ffe3f84f608931578cefc63065946421e081a7d631c32", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 155288, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1ac6545dfef88e72a8f5f0f9e7229e5ac9f63d71943cfa9f17dd97d6316df1d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "68f0f97b3922de98d819208836b14670106b7fd3847bb23098ed034e08c437c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "a55dc6d42323673f0a8f1dd45171f9108556036737b68a5bfd17c69bc01e0bdc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "514510d84794b2b6c1cfdbcbbb0c033710a471286aa05495c9f04491ee9787d2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204056, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256_s2_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "ac5a9df23684a9f86cd396ec1f061a26e0a4788641b03f0f5842775dd8bd1c45", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 204056, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256u2_s2_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7386e8887ecf225147a8a45435ce1b1db183e1025f68c14f2b0bc1b5fc6eefb9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 2
, /* mNumStagesB */ 2
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 2
, /* mNumStagesSmemSfB */ 2
, /* mNumStagesTmemSfA */ 2
, /* mNumStagesTmemSfB */ 2
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c0e50702031a641866325f123da38448afd0260df798d178219e1a1eadaba6f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "4e614bb330626af8ef32e9d5c6cbc4322b55a47b6f1ed9c45cd4e5ab9134bdb0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172688, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "29b92c8955560fc4327d5b6d057a369f10df6fb7d45abee06f8ae144a9a3fc8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 72984, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "954b7047b76130546a2016626f253159d740efdf764615626794f2e8e53e6384", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "8c210e6d6edd4aaefb99d2b4d0cc1f9c8ded0d102ef5d3cec28a769f12196dc9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 73368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "4eb981ba7a87180b1cb5bf71989b82629602fc81fbfa25bce7db5511f0df19b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "4ee954094156a3a41b7206b64d6b62b4b0501bdb56ef53877ed79ae33032211b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "c32501d326c79e833ae45681e02c30e4d9d24aa37732699888f0ff49615239f1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "facea89440e73bee8e9bd4bcc75aad4eb82bf5ad37ac382183825dd891f64f57", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 70928, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "e228a6cb41b2deccb6a4fafca6ebe8529c97e32c65013e9a688780451b16cfb5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 85264, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "ed24a931d0232610790467bf48da2b441de46b8d41b4c3623a1bf212240bb3c9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 71312, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "4b783a89dcbfc190ab31f90fcb17f906b74163155992f1836f154642ed30288e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 85648, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "ce10660eeb52152c6f91c75b284951168cc3d7f45299a37a02e2021b9cbdc1ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 113936, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "b4b5bb0ea4f87b27e2a0fb184b17dbf4a0fa928e97ca15b9421c53fa2abb90f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172304, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "d54e351263abe2552db1eec55ad4238cabbc3f0ff16caa07c4e979295d3d4254", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114320, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "48f09ca5389944f4cee8e67d75f83623d1f2bcd22a2c9210a571ab3829633421", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 72984, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "ff86164b01de3ed18683844033c25a5e4fa1cce453e30197a87d72e0a3721a81", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 172688, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "f0aad398b9574edaa48f1c32d40db75187f2f527690292dbeae77894a1f82e1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 73368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "19aadab09a36f1c6a09ef96bdab63b880d94acefbe38f342088d2b1ebd2707b9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "852c2ebfb68d13583aacac556d3f58b5b51e299f8bce9f66b30b0f71bf10aebe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 89368, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "91fd96b4b505300bc0d8786fb6a4e2d035029dfeaad4cb140f0fd80ed756c832", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "69e09b9351bf93167458aef283b3c0557fc3099c861e8ac6249edb054c9555d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "7b47da87f61950a1c465a6aa8d5aead68aec729057e71ed178cca55674b72367", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "2cfe221917fcf90ac2e6132925a67bb3a5e4d361e0a83ecc8358f9edf28b9dc5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "f9f1ac1bc4b098cdeae082e87dc0e3d18b2c016ce3d9adfd2d75001a33244603", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 89752, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "3f86c65bd2c0ba8405bf79563654a8cb36f3c0d9a8fbdd27f7d09d90474e0400", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123160, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "5dc27a0778c48d7e9ffb7bced0c1d4e28fd43fe8db36c35dad5ed2f726ccba6c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256u2_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "5593a59eba99d710ca540fab84b8a810d8df8e2bf12c30c1a9dce82c7f5e3fdf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 123544, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_TN_transOut_schPd2x1x2x3_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 512, "e6fa4853e103b7bca28f5740567a90a99de8b594a92bf909cf1736478a5726f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(1)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189720, "bmm_Bfloat16_MxE2m1Bfloat16_castBfloat16_patch_Fp32_Ab32_t128x128x256_s3_et128x128_m256x128x16_c2x1x1_rM_TN_transOut_schedS_biasFp32M_bN_rgTma_clmp_dynB_sm107a", 384, "aea2c5a5a1bfe2633c142fee6fd0d7b2dcca6952b63888ef657e2132c92581da", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(1)
, /* mBlockK */ -1
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17826828)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(0)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 1
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(0)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "534b03d4a80319739a2e74edf37112e47fe6a1fca6334d4129afa2b8cf270214", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "8f85186e196d70069d3e89d18bcec4bbe7a85cfe7f2b815ad6bd01a927900cba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "670a289b315e889df0b1aa13c93615712a55be2186ed777ce9d7e97113d28365", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "3b2435ac6f365db0ce611d1d503fbdd488f1edfc2739856d9808f419f7609f8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "dca6679452d1540c26761a66f786c94c4d6f2f2025e5babd02d3d5b41355f85f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "b34f2ae1bc036e9aa01b4b582826650c6fedda93b98e22b91ebf2da035c3e359", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "e2833c37a245ad5de349cd6cc1faf9b78fc091a30e62c6eedf2dfb952dc9ff58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "953e1ef1ff7041c1862740116788b26b1e7d6b5fc79b8d17b0283cff19ea8d05", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "3b29041e9861e256effa00edd9079ee1749a15d1f8c98fd10519fdaea60c3b2a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "88511345db1371b34d86d315e6f1ece9758afc048e671ce098dbd006832aec37", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "288ce99f43feca713eeaee8104b17b76cbe0921e994bf949b35737da5882eabe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "d84ffde92403aea331a6675846aa8bf2caa2dde3838aba9e2a9cfde3b4ec3b0f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "ca800c6a7df8f833ec4da49a1de0cb923a6e68180d31b047f61abfba8434877f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "be3db64397808df99a3ce134ace98740708d5477bd78c161678830b9494389b7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "4707baf934901a89bb28730d0393859f688cb3e6cef7ccd10ebdff11371c36f4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "34f8b12315679a646b496a52f29cb90684d2766b6ff818183b22043b6294525f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "92885feb06665dd3b168a569da04286dc8fb228a73a7fd404fe7f893191e1aef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "9714032e35e3818cc273f8d337871c853fb85d67c294f527a86df9ea4eb24396", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "abfb2ed99e50ce2391a6ec6bdb14b4c642bd368164b80b706533f93f0c483a8d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "7dc14ce395c61a81ba219f2b10ca6a941334b1a92422fa25ce9cf16259951be4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "9dd0d0821cd20e19a8ca351487bd6cef6ee0001d206284e2508b65a7a287cf78", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "77acb87c6a2ea951e1ac21f51822e6e7e41582cfd08365a94c61db4d02089687", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "f52d4b4fad7a76fd43c568bca20c38ade9f6d162d9f99e391abcd85ca2821710", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "32a129fc00c3421caebcad1036424d7753565197bebc3af37ee139b8d6b4b269", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "746b62840fa0da04eba023482b7d308a5f11ac717803f62ca76d0e71816ffe92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "f2f388ab4e29ffade5d1a456b6262fa8d3096a82aadce0b22759cf125c2753d7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "1d0b70af82507a01bd9dcabb64f737304673439b2f2a58f6ca6caac334e40a15", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "0f8742f27fdf4e278c2277a89b88ce239368263eef10038177afb81ca8608b6e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "9901a006ff4e0cad6307185d93f2f88df8d5d99946b25a6dd92d4e273bd5d245", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "341ea2ae20d37ae9072a6553dc982ff9eb072338111bfc7ef71e96921814c0de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "29d107cdc362fe1fa491711d4ce73d57c31e7e65aa2cab30e4b4b714e3d5a130", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "d21d00abd00bad247f67bc93993bfb6c1adc182a7ad1b0d6f8116fecaf15b06c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "dcff6a2cde2a39b8f8abe1cc45935129891a7ebef1f23ea096650ca0f9cb557e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "34888b73c50d5c17ce193ee34903b8ed21bd9874e94ccf1e9d551ab589734483", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "82cc3b4372d83a4589ab00f45767c0074221cffba371c9e50512dadd998dfa04", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "ea25c141152e555b14bb3a65ba39661f57b775cfcae8088e462e19b819d44bc4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "cd7ceb6a19b83c1db01ce03d793d98bdf9d4a06ab370600174402c4ee7236a88", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "0daf17b75558bcc4293025e0d644dfb50aac3c8131f92c499c58151720460d90", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "a30c5d3345fd3e5f097b49f8231e7c171dad66ca65627e5ee67a58a6bcdb0692", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "12e0ca7ffe9abbdb99e23986503b76db07a7de00548b4e942b4064657ebe0e1c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "85a13214782e4b29bcf0e879e6ed1be2f9f70c28ad9a753d0ea9aa5203fb6034", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "9a81715ff802b21dd001eceae82ba67934694a725f9e87562399f0d746638f91", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "aeeb27cc81e853b8c4cd8f1dce32c1a9f28fcabf585e0bf5977f33553d22e539", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "5d7f0239fb6742f6a4c8438f5359df809b82697120915ce05d3c2429ddff8578", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "f347a3b8cde1779f278864339f8834f017eb45d87ad093a8fb6e6d0e045496e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "217737ce15e78768e9fa3361eeec4cff47ac5da3333479b3ad28a4ebe58cb051", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "c7363a0eb659bccd3773b94e422453e895f054eed54ef5cbf7ab2b8062c57d92", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "9e0e1db6e31703c312873280e1505a726053d517e68e3a08779962606df98046", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "db3853a03910e4771e804675deb89aaa8c60b33cb7dbf65bb98bfebc2a1823a6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "c2a46c7e13db96dc5944133ca45e2e6bda63aa55222972b6ab671ad2f6eade52", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "fe7a8888f92819a9a5120b065e053dbd8f8d4bcbcda04b56752bcba5a235c647", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "4f634f1488aa37773faa84fa7f6cd40476ce4f09c292a5e8458277c4f448b37c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "f51415e1548375d19ac543030c5e4c9bb5b9e4d6318398bc412677a31b0b11e6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "7f6b54368edd000592454f5d7c8032fd24ce5a135a09087f1314087fa1fe60d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "240c184662923bbfa5f456315074a78117a22af7b3035a17a075b400989f8259", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "7b0f269b2048aabb7ac6355c8786e70d98acf826f3ddac2e70eae4385d6acf17", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "aa92c4a16ac2c8db9c7ff4f2e9671fdb8c2935dca4440fe78c6e05346bda5f9e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "26b1612f274070d2699a5f94b84e0cee0575339768527ab430f54f572573a5d5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "d0737cf9a5337818349400eec573838dbc24eef80df09730bb89cbab62169c54", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184592, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "73e4f61f0b3d0e3ffe9ee1a72dbeef419abce203bc77e54ce46d3be06914957f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "26a54e5d44a6e68654cd39ca25ce80f27f24b7fdd9f81af1927f6348ce78330b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "b38e9765a7c18dc0c13869530ff60d31584660c48e2a8f50697ccd0ac65c8e27", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114960, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "f7956e2adace3c4b3b9b87cec33a873e7e20c8ce8434e6bd53b73b657e372254", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149904, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "f3635d4c469db6b418e7a884300d57f7dcf8ce5a9f380795075d30d41f0a02a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "6463424dc422e9510e68c4bfb1b953ad6f8dee5c3a45c782948ba86a4929d5ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115088, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "9bac4299685025c92ca315464bcf4a5997cf87674bc047c2bedce3125c372d47", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164112, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "3b3d01e6d31a373a841a0faf4b91892ebe8338099bc99e8ea474b36d6ffb4c76", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "d0904427dad07317f460953a0f72b6e525d18361b818d04b290c4e35874628fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127248, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "7ca64df2ac933e694a5ce2eb2bb3220e04d9050210f081e9360a2168e2fd19e2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164240, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "04b993179c922c44e4a59919c05c41c54878010902c380493922089a9b4a492a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127376, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "b0c1baf922d46008a49aefd5c2fbddb55d7860a2691bdea79b8e5eee8e4a6a94", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "d522d628039e9841ddc494f9bbf341f8c1a21cf8cb613e6ff339982a4edd732d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151824, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "c730680210cb25b85b4647833ac503a86a0ad33e446f317aecd5c669c9ead15b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192912, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "4806e91c3322a8b194f4c825a989e4c55f69093170a78b72f13d48f884de38f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184592, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "7b10285d23e45e1c9f16a77373860757f47e4cd4fa4ee1bfba26802434856ba4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "46b960f135195da8f6cc58fc2dc36ad51dc5124c28fe67f37597bc5ccab7e54f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "e12624b1c337efa01964d141e7403a57f8da25f35eab27fb584821ba7747db6e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151952, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "a60c99ee3d2cf260fb8bab0d0ebbdd18d6f23625194f235fa01b94aeecc72e5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114960, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "8a91b3c8290165397e0c890458f630b608c775c777c931b981351c725b2b4b71", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149904, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "58deba8fb863a7941035acdc881d423320acd6b3b234c0efd9e2c78d002be365", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115088, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "21a4d2789009fddd7d89b6e2e0e599c8ae92388c185d8fcc208f30908e9adace", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "5053aa84d14b167391d61a95af4895ca6be33f6a6f9966d64aab51c6b80db441", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164112, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "315646935aa5ad062a10bcde4e0ce81a59112aba251096879f0c300d60b5f198", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "73ba56cfb7ad97b12a61a759b42dd92c482d266d916d1e3c0f17f1032a059bed", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164240, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "b432da80580215611a897fc75244272161b948005c9b862ce6959e8b8d9d01f5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127248, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "e2142acb93d40078d0ca19ad6e10c34b00e15aac3a7873e777790a6a0fffde7d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127376, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "618b0ffd1c47dc5667a1d20bd456e3b6fc6f2b302d5374324ea3ce42c0af1ab5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "f09673cc813e9e57ff7016ac790f969d3f341137e80f2f8e19d6c2b96e47ac59", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151824, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 256, "7adfa23997a9a0008c84cd29169ccc5d8ddc65009ba389b44ca4f78b1c73bd46", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184592, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "00cd2dc7850994e4674658a2bb7ea3b88c3215a26f12e0cc20a159aabbd5a7e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "49d75b7599cd5b85bad0665f2c325d39247f8c2fce302dccdee8032569790d7a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "5b03248192936d401da6a40ce60214fb7a268756bb631d943b3853f8f9897b0b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192912, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "c131011973f64ec6fbe826718347e3699c2b79ade6199a9a950617c84f36701c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151952, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_ldgsts_ldgstsSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "574bb076c9e0068049db0cc49f77a90c462d543182aad927d71f6bfe59740577", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114960, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "5ad60feaeb49db12929e5441610e5b379ebb62527595531a541f30f418a11462", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149904, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "a2f8cfefaef12b1a92daff8e3d150def9a195437721ae4095bf5b0cb39bf93ec", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "7dd125e74425aa946dcd5bc353be18a7f54aeec26a10bb1f2ed3f5f42008af1a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115088, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "8d2fb8afbb8711a5d8eb68a8132af4ca3b3968b4aa4b10dde2a7ff6c9f66aeeb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164112, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "59e863ff3d7157fb62d8a403a794b5ae731e1f2fc13d70121a3d64cec8adba61", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "864ceae84af90417730484dba62f519b43bf6ce7609bfe897f03a6f2db86694f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127248, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "3ef2974ca7cfbbb1773de68b4539282ca65535e1e1aae61dca10cbcaa5bcc697", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164240, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "615833e73bd87a768fcf52b048fb0cf2e9cea89885d6688b09dc699137d0df82", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127376, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "7dbc55b5e0ba03da4dbefe089534cb1089cc9a91d51a0f8634a947083cc469e3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "33c7966bf91ec1b899977f57b02110875af37767628588777fcd69b106c6eab1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151824, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "6d4633f9a63bf7ab9b977a208f69fb98cf545940bbabec056fb5668881fd1338", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184592, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "d1f6926abd07424dfcd13b11106e744354a14b3b89d3fd35714f9f4b8edfd47a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192912, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "9fcff3531732107fc5820ae93f1aa033f2cc7fce675b33e483e82ed58c7ba8e7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "16775960de63cd3ca647df002882fce576eea9ff719074a74de2c4dc39d3dc9b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 184976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "e2cfbc77c49269af63565a071a7a3744edb2c1b99e5fbc5fbb6ecd4a50f84f57", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151952, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "efcc9d3bc26acb59a781bfc49be6153c68d0cab4974bf7fabac840945cdd8953", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 114960, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "0c9a89e43d5b14f2573a4a5ab67f263358f5dc0144138cfddda0274ec892e4ef", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 149904, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "0b7429df68b1530a4d291b70b29b9a214976b749d5ba0903f53c6447c55654ea", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 115088, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "c09351b875ff954e90baec78a73764e7a653e95fa8730d70a3282db411ecc8a7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 200976, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "9cdea851f96497cf4615bf666ffa92dfa31f940715bc4588297dfa8158418449", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164112, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "c8babaa7a70559ca0c63dbc59bf25cd0f515623aa4f0eb07f9c12db3563b7bb7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 201360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "2eb86da086cb2b9f9d70917b1362b3c86e3f5b187b1221271b4ccef4933c1829", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127248, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "cbcedbecd242f3e783d3f5ed441cf3ed927326a54f53b2f1fbc029f2cf4052dc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 164240, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "87222d36f7359f00eef6e7083c9facd22c339a737ddccb3e24d85f7f94daa68e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 127376, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "08ebf77795a3c65736c5f01dda4706481523047bf852e6dfd6f4d8dc3abf3499", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "1fb70a36fd4f6d1da8a7b20c795c18453eaa62e6b466f22d2aa42456f8d4a780", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151824, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 256, "7acd99acb0495ee6aabd117e913c21fb8639addab9a710f721cc5fc3f9850a58", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 151952, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "a74de93147cd7abb9d083afb7bbfbfb9003744854cca7f2984cbc9439a5b3586", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128u2_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "4cd502dd95615c1a6d0a350cf8a00ca1a3607056e208838bdd3ee82c1b59a352", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192912, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_ldgsts_ldgstsSf_rgTma_clmp_dynB_sm107a", 384, "297ab9f1e883bc9250d3f54e777826b6d2e112bc8664ac0fc840690d5478c0bc", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(1)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(1)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "1bf22e0c2f6f92bb68735de7ab430458c36b2e2550d01ae2d39e0dcebd0f4213", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s6_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "8bc2744605fbb95b7a422feb8843be790005916468c1d36335a3c16a180053a3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128u2_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "cb015083be028feb2401d12ab5f9ab752c9ed91bb5ba7d82565104fd0bb72545", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s6_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "0098d2aee578cca6ece05136c1ace31844d2cb6ce04c945a3e0983120ce65e8b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 64
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 64
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "e95641bbd4fda443a06d85436acf38aa26bf886228c9f8fc40522567bcba50d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s6_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "f5b01ce0afb4812c489a4b8b2b8a75f07b613470e0e368c921953d21a4de051b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 186008, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s6_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "e440e2f05f603d7b9bcb4106932a1458fa5ae1d50f9d2fa8b6e57628310aea17", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 64
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 64
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128u2_s4_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "cdb7675961ed6d6bbc460b3acdd94bfee2bcbfa3e6d041d8fa5a2d49eeb659e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128_s4_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "b70630c97ee34f4b20d56e1b56651b7086ceee76f62c291e3fef90e679bfc10b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128u2_s4_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "efcf11611dbb4231431f54b39ab2601d11b126bd8167163e53866c1106c1eefd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "e19763d7c229a28897b3baaa0e33a44960031ed615e0ee8570828acd7492e730", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226968, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "85a46273f5116ee64cdb664b4e5152d19b81f22909beb3c25da6af8e93f03624", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "cbd3f8d34b948607b8da65c203a55642c9aa0e8c2503c91ca19d11b6e48b9522", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "cb79284d8dc2ff21ff1a0ef25ceb4363d16fc12714cbef5ea4e13e2e1b2c2ee5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "4fec7c29b001c427e613bd4c8fed62ed389c969513572032d1e4d22f6b99e65d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "db1309806fae482b90850455986f64d6bf3bb8ab76448f87dde68a8c8a2530c5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226968, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 384, "0b5d49b6382d2af8ae4e8f8997f357ec09da1bb670652d64dc67bbb97c3d0ae5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 64
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 64
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "45b7cc40aa42e2ba656aea37790d39ee86622ea7307db1f1204a475e6025f116", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128_s4_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "d428ea78f9a1eefda341704acbd7f0d4691905afe2a146259f786306ed081531", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128u2_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "6b15082624f1087d949ea2026b0786aceb2ff7ebacc16c22599390b4aa5655f0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "76355545593427d4b750cfe9fb0d4fa515c5d289dd5843874789c0b63ea30f5b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "b065ca4f8459cff31f1004cacb65363d5b5920f59a58a3f0f52be35c990ddccd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "7f8476668772ed3fc052e56ac4186b8de9b46579e1af8593ef83683e6198ade4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226968, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "c408e3f188ad85fb219813ca9ebbd6847df57b16fbcd8e168ee0f4875d363e6f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 64
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 64
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 226968, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s5_et128x128_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_relu2_bN_tma_tmaSf_rgTma_clmp_dynB_sm107a", 384, "b884c1625e2235df1b8ba56c72830fb7950a2aeb86c05ba265751067d6d8eba2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(2)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 128
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(1)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "de06fcad52ff222cdac225f681d36f42fbd670aebfbbb3a51cbbe9a96b041d4d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "3416897dda1c9693d97f5d45b96400b9b41c588a6e717dca9bcd704a0ce73238", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128u2_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "dee746b55e1c31af4c977ebf419eb6b576749efc77845058dd6acbad8b55d77f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "ef00c78da0049c2c55f5314fa8834bb20b5141a8d793d4b84147cd2168bb6ed0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "c90f12ce1a063a69ef7ffa217cb5f97e4dc9fa6b25d6e70d2b7ba1b8510825bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215312, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "9b412186a184b47cb48a2256e47e86a28eebbc2cadab7ed648541ddd2d1a7336", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180496, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "a191e62a6a6f328a10f8fc13ecbb0ee67a108a2622c53226fb6ec3d7afd7e723", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128u2_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "9abd1f3c7a2658caeb3df0d99c31f40636c9cb6c13a680bbd04dc66332e4f0a8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 215696, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s6_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "4bb5d6930d42c2e06a767710e8c6216752135f82c951540102402ee305493baf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145680, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "83c1b3963029c47c49583f19baf5e21ad06d23cb5b080f006ee408f423dd8cf5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 180880, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s5_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "032fdaef5f1392d0ba8f615815ce117fd39e44917cb390327d79d9d68958ec07", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 229648, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "48a1c2570cbf44555db719b34228756671ad79c30e90f32ccd83eab5b2975251", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 145808, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x8x128_s4_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "e1537bdf9be0d8b1c247a15ff640277d415b11f9e27f4b141c0c1aa7e8f75214", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 230032, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s6_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "95126c19db78151700ba2e8dd6e87cbe407e1253a642e8145b1b529137c1d928", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 6
, /* mNumStagesB */ 6
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 6
, /* mNumStagesSmemSfB */ 6
, /* mNumStagesTmemSfA */ 6
, /* mNumStagesTmemSfB */ 6
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 192784, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "7943bf998e9badbf0d30b8c4e08239a670f9d78033ceb4e63acd9668dcffb6fd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 193168, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s5_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "15ca1e00b03a41e6265bf509215a4137bde3dbc2572213179b6218b166097c33", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 155920, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "b130c26d32d0872f768d018776dbc3dd089fa2d1fd07849c2cd7111743d62db9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217360, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "15f826ef76d77e8ff50572606e7933e900c29b955b024e6e07e89a89474ebba3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 156048, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x16x128_s4_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "a5eee2c7d236d6ce74ac986b06db7702f96b761c195f4ad5941b8e3eec77883f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176400, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 256, "f9a5ec4358b45a5da7aab58d741ea31259e9ff12a052a0a2cf2de39cfa602979", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 217744, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s5_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "6f531ee404b41e777df4b60e874918f5b5e8be7d5fff156a6d093532c6da4f5c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 176528, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x32x128_s4_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "b17c38fc5d2f40727bf8e5de632c36435bc9212be7f2b805e499d81abdff6992", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 0
, /* mNumRegsPerThreadNonEpilogueWarp */ 0
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128u2_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "d06ff4413993450c5b96b4e016c129a5ac08d3ca486076f09fd01cc172404d39", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64u2_s8_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "0685f804e32a61fe093ed7849b8cd0c7ad180eb2902530a11208fe86e5dfb2cd", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 225944, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x64x128_s5_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "45e6f7a3e9f9891f9acb41d1143230822f786a84239b42e98453106bf6080aa7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 5
, /* mNumStagesB */ 5
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 5
, /* mNumStagesSmemSfB */ 5
, /* mNumStagesTmemSfA */ 5
, /* mNumStagesTmemSfB */ 5
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 218776, "bmm_Bfloat16_Bfloat16Bfloat16_Fp32_t128x128x64_s8_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 256, "e409a12ae9ca97f42e8d7e17540fc7cddc2a54698aa11662dc99467b2e442342", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 64
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(1052672)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 64
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 64
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 0
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 168
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 8
, /* mNumStagesB */ 8
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 8
, /* mNumStagesSmemSfB */ 8
, /* mNumStagesTmemSfA */ 8
, /* mNumStagesTmemSfB */ 8
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ -1
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 64
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 64
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 74000, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "997d79b8f82b9d0757e876e7ec27d4f202811be55bf8041c56dd7504f3c1f7e5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 88336, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e120045fa54f3fa084d65b5eaf27db5ec7f3d7ea7798ab1a82dd392750129d3b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 74384, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f1aa3cb68fa8df16c9701932d99c004ba71b23071e2bd48cfab502d96fc5c0ba", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 76056, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e37c1c0eae535ed1f702265dcfb68959d8e79d9d4e07fa4bb4301a497538f4d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117008, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9d0e10fb4c34f6c6aa54fbc95033a6f9452d0629e69b0d9781e0773c7191277f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 88720, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "b28f7cae66d9993dbf1e77f57ec5e53abaf192ebe67c6476f788c4c5f5d67bd0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 76440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8910bb227f6f352dbed519833b073acd1089dd144e5802bc43d0778fd5b0a04f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 92440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a7be30406efe829827d30be3b6945d7d86177730e5601d3f517f2db01df54dc2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4934f8d3e6f574f6c9638159b42d32af0c5669f9edeaf3edae90127e68378e8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 92824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "8688f8d39a6013ef79b7e760e735c935877337406fab34c10e6c0afa4ef5e6eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 174352, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "856b9d82822099a64b2f824155ebbc6653498545b55b7c0789695685abfa5cf2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 74000, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "9dc587a9fc4316973b30647e7ca2ac12546529deae16c16888a3f89088b5b59e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 174736, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "21e09912cf842d1beefdfc92e0d7458a0f7187bb97a7a067769ee137569554ee", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "3da2a31d7541a791a474f2b891576211e39e48013c9148d14e7f5fc5e52b0c49", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 88336, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "40a0da978b877538c51e5392a98f5528f05c196332dd0cca3b99f41728fd95c0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 74384, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "24ee775b7908c999239e61fe1451ae59d9317e5e441ef5540596793d0ed4b43a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 88720, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6d0d4bb9a66eb90f7641c64f73919f44ee25799682972fcfb2adf07d192a95d9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d6ece2324a9dafbd8e37b945d56c2b94268e17f0e044996f0ffb6755e671bd18", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 76056, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e9befe8bb70c92eab4b897d2c1ae0b92bcce5d2e7e35706d8ba6c21d22653f5e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 76440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f6f5795dac2afde5c06616c772cc5e44960c6fd95270b1e2c20289d146ca01e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117008, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "fcc681eef52b74a95505f6d10c753acd591e66d0c0c8208828264181e5c75903", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "f188ee8fdf342b24ad8dc247ed1ba9257e002c594df57bf22095837fe87f3f65", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 92440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "de6378fd070b930b3a14a575f87d84208dbc8277783779c045440e3c067a7a02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 174352, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "bbf86ba311f2cd2497d2861c1fd4caae76e0ed49bbd26704b1fb679b0a3cc810", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 92824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "d43c4a11853658378148f5e1bb52eb2676c948fc51de95c2d16f2d132151b21c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 174736, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "dbaa7dfdef358d68a84dd16370e4e6b5fbd3527c3ae60fe488d752d525c5cd01", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "10c4510fce1de79c5725ffd2983d42858304a7c2d0fe6c52aef5a8660e071168", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 78096, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "071b2310f4ddc6dbcf641d69e4b5eee007c863cee48517a6f4db2d39ab3c1f4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "836738aa626840d020c30ea98c8fcb5bfd7c00fe9dd456e8d7b11f81ecd5b984", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 96528, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6188ae44113dcabe6b7d6c995ed432a35731d3a2ab6811bb47c1bebd45f348df", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 78480, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "94555e43652328253ce163d5fa837f7599124c363c928f8df99085da08ac7cd6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 96912, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "de35e16e962ac5e799d1bf5d9ff7ca1ba973c87c2385050557c3641f2feeb751", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 84248, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6d4dc99af97e9ce032c3fa9dbad38468164211bf0282253f8b94d29a9af4d1ae", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "21ba05c681293fc35b466ea948bbee9455ac143913ff91b78dbc689cb02d6a3d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 84632, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "94a3f01a70373476f2329fad83904cf4a8f2373f91e8a46e3d00d4d70ab11d4b", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "19f8b1bb102d1198b0f4e0911447826f64523477efbe9c326a3d2f0e8c3b97de", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133776, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "7da6c5efc06216757f5d56cd23a4740e0e6ae0700ce62ca52c1bf9aecb5aade4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 109208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "958f04c55cadbad21a3f2e3d08a7aadbcc2c6d5d024d8bcd9cfd06a6e8f3e2bb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207120, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1ce77e548a75616236abc3355606fd43cc71b0594963b524da1829a0c815a035", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 78096, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1c386827a6cc6563d37a04c6eba7b1e5d12287e4f9c8736b662b82e6e884c0e4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 96528, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e817f3709aa8148943928acf5d64a9d70a9c06349533bbc78e8e0385887bdf8e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157976, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "addc886b83a8dc07264f2d34327a5a03eda2742f19c056b9d1442e24fc7472fe", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 78480, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c7f3d1963a9ad798eef37a6f1b81c02dd1bc3113074709d9851f2d489c077059", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 84248, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "c5a60eabc5553cc4856db920a0b71b66bd7e7772437a94aab7393bbfdaec7490", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 96912, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "2b27a236b9c6fc2712126b938041e1e61dc5a885f852dd68bb7c1c8bf9525afb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207504, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "00f1ea588560341a174fdb33d65bfbf765a8920f4927d098698c62d5e6c3d9cb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 84632, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "dea4eb4be118a080160165dc4babc2999f4c22e98ca03bc5749f6e877a247db5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "27830a8f48f3fb22e91e6e7367655670d5489aa4814628995acd5f0b4636ce0a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 108824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e9c562268456b41d93160f98c14ec02d679cc3b20722492b25c92fbbf2dc88eb", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 158360, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "23a5d8359b3cfd5866f79f55533be2a405cf0b2e3e036dcd1698433c695f1f8c", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 133776, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4456365f0f81ec132536ae659ac312c91ed3b5ac03fc71bc312afb8204cb05e9", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 109208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "a163101f199bdaddd483d8459becec8b133205c38f58e4df00d7daad11d859aa", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207120, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "5acb24286f76013c9b6cf6b46f8c6b9e7415f0e8f2fdf066dca1894482411b5a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 157976, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1b221349c0d5124b6de8f3961c0bb880d45f50b6db2d3f594d18d849f5b0a9c8", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 207504, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4406894f042971c64c50e80190076ab142fae452b3900c3f5dff3f03e96d61bf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "1a8ca4c6a7dbdb383c6c1a92ced8ddbc0f244306aa1352039b5f4cead2b1242d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "4f92bbd06f062f37cec24fc868921c105aab409bf5e3b4c962a198a8d853a31a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124568, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e856937f1f2d4ad1c6b19a45834fc2c8c71ff51c38138d437041fccdc8aeecf7", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 74000, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "733e9744633a6665728d965fc5c7e701cf248df5a05247d1f7e7207ca85d7334", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 158360, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "cf84b9c3ede3ce2c9d586f142ad5b8311ad5368771f89a3f217411f77f87e4f3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124568, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "67b0403b928d62c04ee6ff249775db54e44197ba3f0f91317e4980aad9b8848e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 88336, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "9b5ea28fd9821ba29f3eb41329ba267e0dc6f37ea9c5dc53f4e988ee3f16b61f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189976, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e359556ef5c5ffba4d9356c54f400d543cf5f3a5992e8f8276c0a2e390cd0335", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 74384, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256u2_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "474ca8ac119d120a44b918c3eaf00c23b65eb3f91e06e1f0c2598eaccaf2fe02", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 189976, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6cd3cadb7e48c10f418d784bd1515b8fbc0e7eb44c2c646309c01e711c197372", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 88720, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "f27a94ad9b3b1665bd9e0e9a1be7f11ae808adbfd80351850b91fc8f7b232996", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 76056, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "081afd519c04df79815e411c42cd904b25ce84b0b3bb1253b79d0c899594bc84", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190104, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "6df6af170b6806b9d15f63fa02f7a3f2e48988e8e9d022f72704bacb3155045a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117008, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "8eb8ac35231c1c4e34011f053c8178eab4c59b34ddf0e9eb3224c513487690c3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 76440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256u2_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "a7a9782108822fb26e35aca3c32c6a59a4bb9fabd5cd8e4ce41c8a8c1b91e36f", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "02e88366acd1fed3ba8bbf32e3c3b743c759640d81946c8305194f4c4d7668f6", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 190104, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_biasBfloat16Mn_bN_tma_tmaSf_rgTma_clmp_swiGlu_dynB_sm107a", 512, "e73f4eb82ef4c9369f7b68d474ec5f408e72293555224a39d48b886e62067cc3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1052672)
, /* mBiasType */ gemm::BiasType(3)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(2)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 1
, /* mGridWaitForPrimaryRouting */ 1
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(2)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(2)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 92440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "bc4d873bf81a26187dea2f3b38b3f980233866f400795417c3fa9a7c8e75c843", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 174352, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "ca790f441ffd80e14d9726300217db0efdcc669fdb446319b749feb35ad324b5", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 92824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256u2_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "e3725294d0a04f81c5c12fb1ee15d2e623507420bc354c0d124ba75c54ac9381", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "4ca8681aff92a4af2a4ea86995e7a42818f6499c43430c37765cd9d242da68b3", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 74000, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "d7f69a5f71e17b253b84b55fbcc8a46cbddc1194135224debc2915d7f0eb1c97", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 174736, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "60b6889c62ea021daa845668320ca376ea699faeae4d01f5eb8b3e184cae6576", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 88336, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "c2532f97be3e3597834250f1f715ef6f02a5be1b2aff8358603406723553d687", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256u2_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "4ae01bdea2f32d67dc86c1455e7ec77c0b3371a89b204be9a588c09f1da1348d", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 512
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 512
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 74384, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x8x256_s3_et128x8_m128x8x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "cf5374a53fbfabae4246c37c565ba022025e4d3c4190421e6e933485037c60f2", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 8
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 8
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 8
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 76056, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "46997a535311bec39c60e0611ff406d88caa3a0eb9c9ec1e410407c891a0258a", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 88720, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m128x16x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "6b3f6c423768356f6cb58b2778c5d6b0df72cef6f46a614df26dd3dd7cae8b09", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 76440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x16x256_s3_et128x16_m256x16x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "bf5685e41f2a1606622dcc249ccd64b3ceb2d353f10879d579819ece07f09e13", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 16
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 16
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 16
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117008, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "2ce8eb6961449480d8fcffcf57f64bdbeb18d939585b7486780b9800a48ab6cf", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 92440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "24970d7033b0f2c02a223e41d199943c883e91e2935cb8ed5a8d18bb176979d4", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 117392, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m128x32x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "f905fe84a1bbfad952b61fc4352f3be8aad098eb3ad63dfdb4485af61d6a16e0", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 174352, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "f6dab7c42231b0fbb0379476bcdf0b598c8fdffb1c1e32dd0485da2d5a333890", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 92824, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x32x256_s3_et128x32_m256x32x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "989fbf9564a958a7d79b8ebe6f542d69440265f081e4dfdc21f8d8825830a15e", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 32
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 32
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 32
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 174736, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m128x64x16_c1x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "de6ab1283f3b0c605c5229897a62a137a7c525ae9c1c83149efc33895258a621", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 1
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 128
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125208, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "efa31106dcc77fdd82a2b6b9668764ad4c0671d0b92aeed42f920906a25d4664", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "cda30e9c202ef559ec8f1f8723a8ab48d1009a919e13ce0c89643e51923075af", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 125592, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x64x256_s3_et128x64_m256x64x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "24c43170c90a1cbe99106c27adf178625422b70b85fb350aaef54992c4919967", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 64
, /* mMmaTileK */ 256
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 3
, /* mNumStagesB */ 3
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 3
, /* mNumStagesSmemSfB */ 3
, /* mNumStagesTmemSfA */ 3
, /* mNumStagesTmemSfB */ 3
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 256
, /* mTileM */ 128
, /* mTileN */ 64
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124440, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schedS_bN_rgTma_clmp_dynB_sm107a", 384, "fe5d9745d21d3d9e744ebf1c3ce393e880a68f4d4c558d1a138f4985823980e1", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 1
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 1
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(0)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124568, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128u2_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "89bf088324ccba8f0c6fd9230c8093a78b6c6d24b1b54498e55eed4b91c0ab63", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 256
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 1
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 256
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
{nullptr, 0, 124568, "bmm_Bfloat16_MxInt4Bfloat16_castBfloat16_Fp32_Ab32_t128x128x128_s4_et128x64_m256x128x16_c2x1x1_rM_BN_transOut_schPd2x1x2x3_bN_rgTma_clmp_dynB_sm107a", 512, "380e0f25615292dae831d08e71d0abcb8980059674107ba943918b6d32777b28", "", nullptr, nullptr, nullptr, 0, { /* mAllReduceAlgo */ gemm::AllReduceAlgo(0)
, /* mBiasDtype */ trtllm::gen::Dtype(1056776)
, /* mBiasType */ gemm::BiasType(0)
, /* mBlockK */ 256
, /* mClcFastDrain */ 1
, /* mClusterDimX */ 2
, /* mClusterDimY */ 1
, /* mClusterDimZ */ 1
, /* mCtaSwizzleType */ gemm::CtaSwizzleType(0)
, /* mDtypeAcc */ trtllm::gen::Dtype(1056776)
, /* mDtypeA */ trtllm::gen::Dtype(17892366)
, /* mDtypeB */ trtllm::gen::Dtype(1052672)
, /* mDtypeC */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaA */ trtllm::gen::Dtype(1052672)
, /* mDtypeMmaB */ trtllm::gen::Dtype(1052672)
, /* mDtypeSfC */ trtllm::gen::Dtype(1048597)
, /* mEltwiseActType */ gemm::EltwiseActType(0)
, /* mEnablesEarlyExit */ 1
, /* mEnablesDelayedEarlyExit */ 0
, /* mEpilogueLdtmDps */ 16
, /* mEpilogueLdtmBits */ 256
, /* mEpilogueTileM */ 128
, /* mEpilogueTileN */ 64
, /* mFallbackClusterDimX */ 1
, /* mFallbackClusterDimY */ 1
, /* mFallbackClusterDimZ */ 1
, /* mFusedBiasShuffleMode */ gemm::FusedBiasShuffleMode(0)
, /* mFuseLoadSfTask */ 0
, /* mFuseUtccpWithUtcmma */ 0
, /* mGridTriggerSecondaryA */ 0
, /* mGridTriggerSecondaryB */ 1
, /* mGridWaitForPrimaryEarlyExit */ 1
, /* mGridWaitForPrimaryA */ 0
, /* mGridWaitForPrimaryB */ 1
, /* mHoistLoadTaskInit */ 1
, /* mHoistMmaTaskTryWaits */ 0
, /* mK */ 128
, /* mKernelTraits */ {}
, /* mLayoutA */ gemm::MatrixLayout(2)
, /* mLayoutB */ gemm::MatrixLayout(0)
, /* mM */ 256
, /* mMmaK */ 16
, /* mMmaKind */ trtllm::gen::MmaKind(1)
, /* mMmaM */ 256
, /* mMmaN */ 128
, /* mMmaTileK */ 128
, /* mMockAllReduce */ 0
, /* mN */ 256
, /* mNumEpilogueWarps */ 4
, /* mNumRegsCastAWarps */ 112
, /* mNumRegsCopySfLdsSttm */ 0
, /* mNumRegsCopySparsityInfo */ 0
, /* mNumRegsPerThreadEpilogueWarp */ 160
, /* mNumRegsPerThreadNonEpilogueWarp */ 96
, /* mNumSlicesForSplitK */ 1
, /* mNumStagesA */ 4
, /* mNumStagesB */ 4
, /* mNumStagesMma */ 2
, /* mNumStagesMmaWithinWorkTile */ 1
, /* mNumStagesMmaAcrossWorkTile */ 2
, /* mNumStagesSmemSfA */ 4
, /* mNumStagesSmemSfB */ 4
, /* mNumStagesTmemSfA */ 4
, /* mNumStagesTmemSfB */ 4
, /* mNumStagesWorkId */ 3
, /* mOutputDebugTensors */ 0
, /* mPatchF2fp */ 0
, /* mPerTokenSfDtype */ trtllm::gen::Dtype(1052672)
, /* mSchedHostTask */ gemm::SchedHostTask(0)
, /* mSfBlockSizeA */ 32
, /* mSfBlockSizeB */ -1
, /* mSfBlockSizeC */ -1
, /* mSfLayoutA */ trtllm::gen::SfLayout(3)
, /* mSfLayoutB */ trtllm::gen::SfLayout(3)
, /* mSfLayoutC */ trtllm::gen::SfLayout(3)
, /* mSfReshapeFactor */ 1
, /* mSparsityA */ trtllm::gen::Sparsity(0)
, /* mSplitK */ gemm::SplitK(0)
, /* mTileK */ 128
, /* mTileM */ 128
, /* mTileN */ 128
, /* mTileScheduler */ gemm::TileScheduler(1)
, /* mTransposeMmaOutput */ 1
, /* mUseCustomizedMma3xNvFp4 */ 0
, /* mUseCustomMmaSchedule */ 1
, /* mUseDeepSeekFp8 */ 0
, /* mUseFlexibleClusterDims */ 0
, /* mUseHoistTryWaitForCustomMmaSchedule */ 0
, /* mUseMaxTmemOverlap */ 0
, /* mUsePerTokenSfA */ 0
, /* mUsePerTokenSfB */ 0
, /* mUseShuffledMatrix */ 1
, /* mUseTmaStore */ 1
, /* mUseTwoTmaLoadWarps */ 1
, /* mUseTwoMmaWarps */ 0
, /* mUseUnrollLoop2xForMma */ 0
, /* mValidM */ 256
, /* mValidN */ 256
, /* mValidK */ 128
, /* mWorldSize */ 1
, /* mActType */ gemmGatedAct::ActType(0)
, /* mClampBeforeAct */ 1
, /* mBatchedM */ {}
, /* mBatchedN */ {}
, /* mBatchMode */ batchedGemm::BatchedGemmOptions::BatchMode(1)
, /* mBatchStrideInTokens */ -1
, /* mFusedAct */ 0
, /* mGridWaitForPrimaryRouting */ 0
, /* mIsStaticBatch */ 0
, /* mIsUniformNumTokensPerBatch */ 0
, /* mNumBatches */ 128
, /* mNumRegsPerThreadLoadA */ 0
, /* mNumRegsPerThreadLoadB */ 0
, /* mNumRegsPerThreadLoadSfA */ 0
, /* mNumRegsPerThreadLoadSfB */ 0
, /* mNumTokens */ 2
, /* mNumWarpsLoadA */ 0
, /* mNumWarpsLoadB */ 0
, /* mNumWarpsLoadSfA */ 0
, /* mNumWarpsLoadSfB */ 0
, /* mRouteImpl */ batchedGemm::RouteImpl(0)
, /* mRouteSfsImpl */ {batchedGemm::RouteImpl(0)}
, /* mPrefetchB */ 0
, /* mUseCMultiCast */ 0
, /* mUseTmaOobOpt */ 1
 }, gemm::SmVersion::Sm107a},
#endif // EXCLUDE_SM_100
};
// clang-format on
} // namespace kernels
} // namespace tensorrt_llm

}

